Fix the bug of Missing dependencies and DeOldify path set in demo colorization_python

bmf/demo/colorization_python/Readme.md

@@ -35,7 +35,7 @@ git clone https://github.com/eefengwei/DeOldify.git DeOldify
 
 ### 1.2 install the dependent python packages
 ```Bash
-pip3 install -r ./DeOldify/requirements-colab.txt
+pip3 install -r ./requirements.txt
 ```
 
 ### 1.3 download the pretrained-weights
@@ -46,7 +46,7 @@ wget -c https://data.deepai.org/deoldify/ColorizeVideo_gen.pth -O ./DeOldify/mod
 
 ### 1.4 pip install BMF packages
 ```Bash
-pip3 install bmf
+pip3 install BabitMF BabitMF-GPU
 ```
 ### 1.5 verify the FFmpeg libraries is installed and version is correct
 
@@ -73,7 +73,7 @@ ffmpeg -version
 
 ```Python
 import sys
-sys.path.insert(0, '/content/DeOldify')
+sys.path.insert(0, './DeOldify')
 print(sys.path)
 ```
 

bmf/demo/colorization_python/deoldify_demo.py

@@ -1,12 +1,6 @@
 import bmf
 import py_deoldify_module
 
-# Add the DeOldify folder to the python search path
-import sys
-
-sys.path.insert(0, './DeOldify')
-print(sys.path)
-
 input_video_path = './DeOldify/test_videos/test_video.mp4'
 output_video_path = 'colored_video.mp4'
 model_weight_path = './DeOldify/'

bmf/demo/colorization_python/py_deoldify_module.py

@@ -4,6 +4,12 @@
 import PIL
 import bmf.hml.hmp as mp
 
+# Add the DeOldify folder to the python search path
+import sys
+
+sys.path.insert(0, './DeOldify')
+print(sys.path)
+
 from deoldify import device
 from deoldify.device_id import DeviceId
 import torch

bmf/demo/colorization_python/requirements.txt

@@ -0,0 +1,9 @@
+fastai==1.0.60
+tensorboardX>=1.6
+ffmpeg-python
+yt-dlp
+opencv-python>=4.2.0.32
+Pillow
+tornado
+imgaug==0.2.6
+ipython