ZeroGPU

app.py

@@ -5,6 +5,7 @@ import tempfile
 import imageio
 import torch
 from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
+import spaces
 
 pipe = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_576w", torch_dtype=torch.float16)
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
@@ -19,6 +20,7 @@ def export_to_video(frames: np.ndarray, fps: int) -> str:
     writer.close()
     return out_file.name
 
+@spaces.GPU
 def infer(prompt):
     negative_prompt = "text, watermark, copyright, blurry, nsfw"
     video_frames = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=40, height=320, width=576, num_frames=24).frames[0]