Update app.py

app.py

@@ -17,7 +17,7 @@ interface = gr.Interface(
     theme="soft"
 )
 
-# Define the function to generate the video
+@spaces.GPU(duration=200)
 def generate_video(img):
     # Convert the input image to a tensor
     img_tensor = torch.tensor(img).unsqueeze(0) / 255.0