Pip package broken: missing caffee

[fengwang]

I installed version '0.2.1-00' using pip. This package depends on python python-numpy python-scikit-image python-requests python-scipy packages. But when I was trying to colorize one image, I got such an error

Traceback (most recent call last):
  File "/usr/bin/autocolorize", line 8, in <module>
    from autocolorize.__main__ import main
  File "/usr/lib/python3.9/site-packages/autocolorize/__init__.py", line 7, in <module>
    from .checker import checker_main
  File "/usr/lib/python3.9/site-packages/autocolorize/checker.py", line 5, in <module>
    import caffe
ModuleNotFoundError: No module named 'caffe'