Spaces:
Running
on
Zero
Running
on
Zero
Update for ZeroGPU support
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import torch
|
| 3 |
import time
|
|
@@ -55,7 +56,7 @@ except Exception as e:
|
|
| 55 |
demo.launch()
|
| 56 |
exit()
|
| 57 |
|
| 58 |
-
|
| 59 |
def generate_video(
|
| 60 |
pil_image: Image.Image,
|
| 61 |
prompt: str,
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import gradio as gr
|
| 3 |
import torch
|
| 4 |
import time
|
|
|
|
| 56 |
demo.launch()
|
| 57 |
exit()
|
| 58 |
|
| 59 |
+
@spaces.GPU(duration=300)
|
| 60 |
def generate_video(
|
| 61 |
pil_image: Image.Image,
|
| 62 |
prompt: str,
|