update.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 import tempfile
+import spaces
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import export_to_video
 
@@ -12,6 +13,7 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
     MODEL, torch_dtype=torch.float16
 ).to("cuda")
 
+@spaces.GPU(duration=300)
 def infer(first_image, last_image, prompt, guidance=7.5, frames=25):
     # Generate the in-between frames
     video = pipe(
@@ -41,4 +43,4 @@ demo = gr.Interface(
 )
 
 # Enable the REST API
-demo.queue(concurrency_count=1).launch(show_api=True)
+demo.queue(concurrency_count=1).launch(show_api=True, ssr_mode=False)