Zero gpu

app.py

@@ -9,6 +9,7 @@ import tempfile
 import gradio as gr
 import imageio
 import numpy as np
+import spaces
 import torch
 from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
 
@@ -48,6 +49,7 @@ def to_video(frames: list[np.ndarray], fps: int) -> str:
     return out_file.name
 
 
+@spaces.GPU
 def generate(prompt: str, seed: int, num_frames: int,
              num_inference_steps: int) -> str:
     generator = torch.Generator().manual_seed(seed)

requirements.txt

@@ -3,6 +3,7 @@ diffusers==0.18.2
 gradio==3.36.1
 huggingface-hub==0.16.4
 imageio[ffmpeg]==2.31.1
-torch==2.0.1
-torchvision==0.15.2
+spaces==0.7.0
+torch==2.0.0
+torchvision==0.15.1
 transformers==4.30.2