increase batch size and num render workers

gradio_demo.py

@@ -155,12 +155,12 @@ def process_video_frames(video_path: str, temp_dir: Path, progress=gr.Progress()
     render_size = 512
     # Prepare dataset and dataloader
     dataset = VideoFrameDataset(video_path, fa_model)
-    dataloader = DataLoader(dataset, batch_size=1, num_workers=0)
+    dataloader = DataLoader(dataset, batch_size=8, num_workers=0)
     fps = dataset.fps
     num_frames = len(dataset)
     # Prepare rendering thread and queue
     render_queue = Queue(maxsize=32)
-    num_render_workers = 1
+    num_render_workers = 4
     rendering_threads = []
     for _ in range(num_render_workers):
         thread = RenderingThread(render_queue, temp_dir, flame.faces, c2w, render_size)