Update app.py

app.py

@@ -1,7 +1,10 @@
-import gradio as gr
+import spaces
+import os
+
 import torch
+
+import gradio as gr
 import imageio
-import os
 import time
 import random
 import gc
@@ -35,6 +38,7 @@ pipe = Image2VideoPipeline(
 print("Model loaded successfully.")
 
 # --- Inference Function ---
+@spaces.GPU(required=True)
 def generate_video(input_image, prompt, guidance_scale, inference_steps, num_frames, fps, seed):
     """
     Main function to generate video from an image and a prompt.