From d3b61cce8afaa725ddf4eea64c96496336d51905 Mon Sep 17 00:00:00 2001
From: Jack Lau <131463945+JackLau1222@users.noreply.github.com>
Date: Tue, 10 Sep 2024 16:04:10 +0800
Subject: [PATCH] Fix the bug of Missing dependencies and DeOldify path set in
 demo colorization_python (#118)

* fix the bug of requirements and DeOldify path set
* Update Readme.md
---
 bmf/demo/colorization_python/Readme.md        | 6 +++---
 bmf/demo/colorization_python/requirements.txt | 9 +++++++++
 2 files changed, 12 insertions(+), 3 deletions(-)
 create mode 100644 bmf/demo/colorization_python/requirements.txt

diff --git a/bmf/demo/colorization_python/Readme.md b/bmf/demo/colorization_python/Readme.md
index b4cebbe4..8e982cdd 100644
--- a/bmf/demo/colorization_python/Readme.md
+++ b/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://huggingface.co/spensercai/DeOldify/resolve/main/ColorizeVideo_ge
 
 ### 1.4 pip install BMF packages
 ```Bash
-pip3 install bmf
+pip3 install 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)
 ```
 
diff --git a/bmf/demo/colorization_python/requirements.txt b/bmf/demo/colorization_python/requirements.txt
new file mode 100644
index 00000000..b56cdb9d
--- /dev/null
+++ b/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
\ No newline at end of file