Update app.py

app.py

@@ -18,6 +18,7 @@ else:
 	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()
 
+chunk_size=16
 
 def load_file(path):
 
@@ -58,8 +59,8 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
 		with gr.Tabs():
 			from gradio_tabs.animation import animation
 			from gradio_tabs.vid_edit import vid_edit
-			animation(gen, device)
-			vid_edit(gen, device)
+			animation(gen, chunk_size, device)
+			vid_edit(gen, chunk_size, device)
 
     
 demo.launch(