Upload folder using huggingface_hub

app.py

@@ -38,10 +38,8 @@ pipe.load_components(trust_remote_code=True, device_map="cuda", torch_dtype={"de
 
 @spaces.GPU(120)
 def predict(image, prompt):
-    return image
-    #output = pipe(image=image, prompt=prompt, num_frames=141)
-    #return export_to_video(output.values['videos'][0], "output.mp4")
-
+    output = pipe(image=image, prompt=prompt, num_frames=141)
+    return export_to_video(output.values['videos'][0], "output.mp4")
 
 examples = []
 
@@ -93,4 +91,5 @@ with gr.Blocks(css=css) as demo:
         outputs=[result],
     )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()