Migrate to ZeroGPU

app.py

@@ -17,6 +17,7 @@ import tempfile
 import imageio_ffmpeg
 import gradio as gr
 import torch
+import spaces
 from PIL import Image
 from diffusers import (
     CogVideoXPipeline,
@@ -196,7 +197,7 @@ def convert_prompt(prompt: str, retry_times: int = 3) -> str:
             return response.choices[0].message.content
     return prompt
 
-
+@spaces.GPU(duration=300)
 def infer(
     prompt: str,
     image_input: str,