Spaces:
Running
on
Zero
Running
on
Zero
lzyhha
commited on
Commit
·
4828266
1
Parent(s):
307cfd5
duration
Browse files
app.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
import argparse
|
| 2 |
-
|
| 3 |
from visualcloze import VisualClozeModel
|
| 4 |
import gradio as gr
|
| 5 |
import examples
|
|
@@ -496,8 +496,7 @@ def create_demo(model):
|
|
| 496 |
return demo
|
| 497 |
|
| 498 |
|
| 499 |
-
|
| 500 |
-
# @spaces.GPU(duration=10)
|
| 501 |
def generate(
|
| 502 |
images,
|
| 503 |
prompts,
|
|
|
|
| 1 |
import argparse
|
| 2 |
+
import spaces
|
| 3 |
from visualcloze import VisualClozeModel
|
| 4 |
import gradio as gr
|
| 5 |
import examples
|
|
|
|
| 496 |
return demo
|
| 497 |
|
| 498 |
|
| 499 |
+
@spaces.GPU(duration=120)
|
|
|
|
| 500 |
def generate(
|
| 501 |
images,
|
| 502 |
prompts,
|