Update gradio_tabs/vid_edit.py

gradio_tabs/vid_edit.py

@@ -8,17 +8,6 @@ import imageio
 from einops import rearrange
 import spaces
 
-# extensions_dir = "./torch_extension/"
-# os.environ["TORCH_EXTENSIONS_DIR"] = extensions_dir
-
-# from networks.generator import Generator
-
-# device = torch.device("cuda")
-# ckpt_path = './models/lia-x.pt'
-# gen = Generator(size=512, motion_dim=40, scale=2).to(device)
-# gen.load_state_dict(torch.load(ckpt_path, weights_only=False))
-# gen.eval()
-
 output_dir = "./res_gradio"
 os.makedirs(output_dir, exist_ok=True)
 
@@ -130,13 +119,6 @@ def vid_all_save(vid_d, vid_a, fps, output_path=output_dir + "/output_vid.mp4",
 
 
 def vid_edit(gen, device):
-
-	# from networks.generator import Generator
-
-	# device = torch.device("cuda")
-	# gen = Generator(size=512, motion_dim=40, scale=2).to(device)
-	# gen.load_state_dict(torch.hub.load_state_dict_from_url(f"https://huggingface.co/YaohuiW/LIA-X/resolve/main/lia-x.pt"))
-	# gen.eval()
     
     
 	@torch.no_grad()
@@ -170,10 +152,6 @@ def vid_edit(gen, device):
 
 	def clear_media():
 		return None, None, None, *([0] * len(labels_k))
-
-	#image_output = gr.Image(label="Image", type='numpy', interactive=False, width=512)
-	#video_output = gr.Video(label="Video", width=512)
-	#video_all_output = gr.Video(label="Videos")
    
 	with gr.Tab("Video Editing"):
 
@@ -202,14 +180,6 @@ def vid_edit(gen, device):
 							visible=True,
 						)
 
-				# with gr.Row():
-				# 	with gr.Column(scale=1):
-				# 		with gr.Row():	# Buttons now within a single Row
-				# 			edit_btn = gr.Button("Edit")
-				# 			clear_btn = gr.Button("Clear")
-				# 		with gr.Row():
-				# 			animate_btn = gr.Button("Generate")
-
 			with gr.Column(scale=2):
 
 				with gr.Row():
@@ -300,7 +270,7 @@ def vid_edit(gen, device):
             fn=edit_vid,
 			inputs=[video_input] + inputs_s,
             outputs=[video_output, video_all_output],
-            cache_examples=True
+            cache_examples="lazy",
 		)