Update app.py

app.py

@@ -4,6 +4,7 @@ from huggingface_hub import hf_hub_download
 
 from gradio_tabs.animation import animation
 from gradio_tabs.vid_edit import vid_edit
+from gradio_tabs.img_edit import img_edit
 from networks.generator import Generator
 
 device = torch.device("cuda")
@@ -12,7 +13,7 @@ ckpt_path = hf_hub_download(repo_id="YaohuiW/LIA-X", filename="lia-x.pt")
 gen.load_state_dict(torch.load(ckpt_path, weights_only=True))
 gen.eval()
 
-chunk_size=8
+chunk_size=32
 
 def load_file(path):
 
@@ -41,6 +42,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
 	with gr.Row():
 		with gr.Tabs():
 			animation(gen, chunk_size, device)
+			img_edit(gen, device)
 			vid_edit(gen, chunk_size, device)
 
 demo.launch(allowed_paths=["./data/source","./data/driving"])