update progress

gradio_demo.py

@@ -187,7 +187,7 @@ def process_video_frames(video_path: str, temp_dir: Path, progress=gr.Progress()
                 render_queue.put((frame_idx, cropped_frame, verts[i]))
                 frame_idx += 1
                 progress(
-                    frame_idx / num_frames, desc=f"Processing frame {frame_idx}/{num_frames}"
+                    0.95 * frame_idx / num_frames, desc=f"Processing frame {frame_idx}/{num_frames}"
                 )
     # Stop rendering threads
     for _ in range(num_render_workers):

video_demo.py

@@ -275,12 +275,12 @@ class VideoFrameDataset(IterableDataset):
 def process_video(
     video_path: str,
     model_type: str = "expressive",
-    batch_size: int = 8,
+    batch_size: int = 1,
     num_workers: int = 0,
     device: str = "cuda" if torch.cuda.is_available() else "cpu",
     output_video_path: Optional[str] = None,
     render_size: int = 512,
-    num_render_workers: int = 16,
+    num_render_workers: int = 1,
     max_queue_size: int = 128,
 ) -> List[Dict[str, Any]]:
     """