Update gradio_tabs/vid_edit.py

gradio_tabs/vid_edit.py

@@ -128,7 +128,7 @@ def vid_all_save(vid_d, vid_a, fps, output_path=output_dir + "/output_vid.mp4",
 
 	return output_path, output_all_path
 
-@spaces.GPU
+
 def vid_edit():
 
 	from networks.generator import Generator
@@ -137,8 +137,9 @@ def vid_edit():
 	#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.load_state_dict(torch.hub.load_state_dict_from_url(f"https://huggingface.co/YaohuiW/LIA-X/resolve/main/lia-x.pt"))
 	gen.eval()
-
+    
 	#output_dir = "./res_gradio"
 	#os.makedirs(output_dir, exist_ok=True)
     
@@ -202,8 +203,8 @@ def vid_edit():
 								#["./data/driving/driving6.mp4"],
 								#["./data/driving/driving7.mp4"],
 								["./data/driving/driving3.mp4"],
-								["./data/driving/driving8.mov"],
-								["./data/driving/driving9.mov"],
+								["./data/driving/driving8.mp4"],
+								["./data/driving/driving9.mp4"],
 							],
 							inputs=[video_input],
 							cache_examples=False,
@@ -299,7 +300,7 @@ def vid_edit():
 				 0, 0, 0, 0, -0.3, 0, 0],
 				['./data/driving/driving3.mp4', -0.6, 0, 0, 0, 0, 0, 0,
 				 0, 0, 0, 0, 0, 0, 0],
-				['./data/driving/driving9.mov', 0, 0, 0, 0, 0, 0, 0,
+				['./data/driving/driving9.mp4', 0, 0, 0, 0, 0, 0, 0,
 				 0, 0, 0, 0, 0, -0.1, 0.07],
 			],
 			inputs=[video_input] + inputs_s