Upload folder using huggingface_hub

app.py

@@ -35,8 +35,8 @@ blocks.sub_blocks.insert("image_to_action", image_to_action_block, 0)
 pipe = MatrixGameWanModularPipeline(blocks, "diffusers/matrix-game-2-modular")
 pipe.load_components(trust_remote_code=True, device_map="cuda", torch_dtype={"default": torch.bfloat16, "vae": torch.float32})
 
-@spaces.GPU(120)
-def predict(image, prompt):
+@spaces.GPU(duration=120)
+def predict():
     output = pipe(image=image, prompt=prompt, num_frames=141)
     return export_to_video(output.values['videos'][0], "output.mp4")
 
@@ -55,7 +55,6 @@ css = """
 }
 #edit_text{margin-top: -62px !important}
 """
-
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
@@ -83,10 +82,7 @@ with gr.Blocks(css=css) as demo:
     gr.on(
         triggers=[run_button.click],
         fn=predict,
-        inputs=[
-            image,
-            prompt,
-        ],
+        inputs=[image, prompt],
         outputs=[result],
     )