Update gradio_tabs/animation.py

gradio_tabs/animation.py

@@ -113,7 +113,7 @@ def vid_postprocessing(video, fps, output_path=output_dir + "/output_vid.mp4"):
 	return output_path
 
 
-def animation(gen, device):
+def animation(gen, chunk_size, device):
     
 	@torch.no_grad()
 	def edit_media(image, *selected_s):
@@ -136,7 +136,7 @@ def animation(gen, device):
 		image_tensor = image_tensor.to(device)
 		video_target_tensor = vid_target_tensor.to(device)
 
-		animated_video = gen.animate(image_tensor, video_target_tensor, labels_v, selected_s)
+		animated_video = gen.animate_batch(image_tensor, video_target_tensor, labels_v, selected_s, chunk_size)
 
 		# postprocessing
 		animated_video = vid_postprocessing(animated_video, fps)