Fix dependency.

pyproject.toml

@@ -21,7 +21,7 @@ dependencies = [
     "moviepy==1.0.3", "scenedetect==0.6.3", "numpy", "scikit-learn==1.2.2",
     "einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13",
     "bitsandbytes==0.41.0", "pydantic<2,>=1", "markdown2[all]", 
-    "gradio==3.35.2", "gradio_client==0.2.9",
+    "gradio==3.35.2", "gradio_client==0.2.9", opencv-python,
     "requests", "httpx==0.24.0", "openai", "uvicorn", "fastapi", "wandb"
 ]
 

requirements.txt

@@ -1,3 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cu118
 huggingface_hub==0.22.2
 torch>=2.0.1
 torchvision>=0.15.2
@@ -14,6 +15,7 @@ imageio==2.34.0
 imageio-ffmpeg==0.4.9
 moviepy==1.0.3
 scenedetect==0.6.3
+opencv-python
 numpy
 scikit-learn==1.2.2
 einops==0.6.1