Update app.py

app.py

@@ -1,13 +1,15 @@
+import os
 import gradio as gr
 from diffusers import StableVideoDiffusionPipeline
 import torch
-import os
+
+# Ensure dependencies are installed
 os.system("pip install -r requirements.txt")
-iface.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))
 
 # Model Load
 model_id = "stabilityai/stable-video-diffusion-img2vid"
-pipe = StableVideoDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32).to("cpu")
+pipe = StableVideoDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)  
+pipe.to("cpu")  # Ensure model runs on CPU
 
 # Function to Generate Video
 def generate_video(prompt):
@@ -18,5 +20,8 @@ def generate_video(prompt):
 
 # Gradio Interface
 iface = gr.Interface(fn=generate_video, inputs="text", outputs="video")
-iface.launch(server_name="0.0.0.0", server_port=8080)
+
+# Launch Gradio Server on Hugging Face Space
+iface.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))
+