Stable-Video-Diffusion-Img2Vid

Paused

Kvikontent commited on May 15, 2024

Commit

33341a3

verified ·

1 Parent(s): d18f074

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ interface = gr.Interface(
     theme="soft"
 )
-# Define the function to generate the video
 def generate_video(img):
     # Convert the input image to a tensor
     img_tensor = torch.tensor(img).unsqueeze(0) / 255.0

     theme="soft"
 )
+@spaces.GPU(duration=200)
 def generate_video(img):
     # Convert the input image to a tensor
     img_tensor = torch.tensor(img).unsqueeze(0) / 255.0