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,
@@ -30,7 +31,7 @@ from datetime import datetime, timedelta
 
 from diffusers.image_processor import VaeImageProcessor
 from openai import OpenAI
-import moviepy.editor as mp
+import moviepy as mp
 import utils
 from rife_model import load_rife_model, rife_inference_with_latents
 from huggingface_hub import hf_hub_download, snapshot_download
@@ -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,
@@ -281,8 +282,8 @@ def infer(
 
 def convert_to_gif(video_path):
     clip = mp.VideoFileClip(video_path)
-    clip = clip.set_fps(8)
-    clip = clip.resize(height=240)
+    clip = clip.with_fps(8)
+    clip = clip.resized(height=240)
     gif_path = video_path.replace(".mp4", ".gif")
     clip.write_gif(gif_path, fps=8)
     return gif_path

requirements.txt

@@ -1,4 +1,3 @@
-spaces>=0.29.3
 safetensors>=0.4.5
 spandrel>=0.4.0
 tqdm>=4.66.5
@@ -9,9 +8,6 @@ accelerate>=0.34.2
 opencv-python>=4.10.0.84
 sentencepiece>=0.2.0
 numpy==1.26.0
-torch==2.2.0
-torchvision
-gradio>=4.44.0
 imageio>=2.34.2
 imageio-ffmpeg>=0.5.1
 openai>=1.45.0