Upload folder using huggingface_hub

app.py

@@ -36,7 +36,7 @@ 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(duration=120)
-def predict():
+def predict(image, prompt):
     output = pipe(image=image, prompt=prompt, num_frames=141)
     return export_to_video(output.values['videos'][0], "output.mp4")