Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import torch
 import gradio as gr
 import time
@@ -13,11 +14,14 @@ pipe.enable_sequential_cpu_offload()
 pipe.vae.enable_tiling()
 pipe.vae.enable_slicing()
 
+@spaces.GPU(duration=120)
 def generate_video(image, prompt):
     if image is None:
         raise gr.Error("Please upload an input image")
     if not prompt:
         raise gr.Error("Please enter a text prompt")
+
+    pipe.to('cuda')
     
     # Load uploaded image
     input_image = load_image(image)