Update app.py

app.py

@@ -26,6 +26,17 @@ import spaces
 # from gradio_tabs.animation import animation
 # from gradio_tabs.vid_edit import vid_edit
 
+extensions_dir = "./torch_extension/"
+os.environ["TORCH_EXTENSIONS_DIR"] = extensions_dir
+
+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()
+
+
 def load_file(path):
 
 	with open(path, 'r', encoding='utf-8') as f:
@@ -63,10 +74,10 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
 	
 	with gr.Row():
 		with gr.Tabs():
-			#from gradio_tabs.animation import animation
+			from gradio_tabs.animation import animation
 			from gradio_tabs.vid_edit import vid_edit
-			#animation()
-			vid_edit()
+			animation(gen)
+			vid_edit(gen)
 			#load_tabs()
 			#animation()
 			#vid_edit()