Update gradio_tabs/vid_edit.py

gradio_tabs/vid_edit.py

@@ -118,7 +118,7 @@ def vid_all_save(vid_d, vid_a, fps, output_path=output_dir + "/output_vid.mp4",
 	return output_path, output_all_path
 
 
-def vid_edit(gen, device):
+def vid_edit(gen, chunk_size, device):
     
     
 	@torch.no_grad()
@@ -142,7 +142,7 @@ def vid_edit(gen, device):
 		video_target_tensor, fps = vid_preprocessing(video, 512)
 		video_target_tensor = video_target_tensor.to(device)
 
-		edited_video_tensor = gen.edit_vid(video_target_tensor, labels_v, selected_s)
+		edited_video_tensor = gen.edit_vid_batch(video_target_tensor, labels_v, selected_s, chunk_size)
 
 		# de-norm
 		animated_video, animated_all_video = vid_all_save(video_target_tensor, edited_video_tensor, fps)