fix: dependencies

README.md

@@ -177,11 +177,11 @@ pip install -r requirements.txt
 Install the MMLab ecosystem packages:
 
 ```bash
-pip install --no-cache-dir -U openmim 
-mim install mmengine 
-mim install "mmcv==2.0.1" 
-mim install "mmdet==3.1.0" 
-mim install "mmpose==1.1.0" 
+pip install --no-cache-dir -U openmim
+mim install mmengine
+mim install "mmcv==2.0.1"
+mim install "mmdet==3.1.0"
+mim install "mmpose==1.1.0"
 ```
 
 ### Setup FFmpeg

app.py

@@ -4,7 +4,6 @@ import pdb
 import re
 
 import gradio as gr
-import spaces
 import numpy as np
 import sys
 import subprocess
@@ -181,7 +180,6 @@ def fast_check_ffmpeg():
         return False
 
 
-@spaces.GPU(duration=600)
 @torch.no_grad()
 def inference(audio_path, video_path, bbox_shift, extra_margin=10, parsing_mode="jaw", 
               left_cheek_width=90, right_cheek_width=90, progress=gr.Progress(track_tqdm=True)):

requirements.txt

@@ -1,21 +1,20 @@
 diffusers==0.30.2
 accelerate==0.28.0
+numpy==1.23.5
 tensorflow==2.12.0
 tensorboard==2.12.0
 opencv-python==4.9.0.80
 soundfile==0.12.1
 transformers==4.39.2
-huggingface_hub==0.25.0
+huggingface_hub==0.30.2
 librosa==0.11.0
-numpy==1.24.4
 einops==0.8.1
+gradio==5.24.0
 
 gdown
 requests
 imageio[ffmpeg]
 
-gradio
-spaces
 omegaconf
 ffmpeg-python
 moviepy

scripts/realtime_inference.py

@@ -327,7 +327,7 @@ if __name__ == "__main__":
     parser.add_argument("--fps", type=int, default=25, help="Video frames per second")
     parser.add_argument("--audio_padding_length_left", type=int, default=2, help="Left padding length for audio")
     parser.add_argument("--audio_padding_length_right", type=int, default=2, help="Right padding length for audio")
-    parser.add_argument("--batch_size", type=int, default=25, help="Batch size for inference")
+    parser.add_argument("--batch_size", type=int, default=20, help="Batch size for inference")
     parser.add_argument("--output_vid_name", type=str, default=None, help="Name of output video file")
     parser.add_argument("--use_saved_coord", action="store_true", help='Use saved coordinates to save time')
     parser.add_argument("--saved_coord", action="store_true", help='Save coordinates for future use')