Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(
|
|
47 |
model.eval()
|
48 |
conv_mode = "v1"
|
49 |
|
50 |
-
@spaces.GPU
|
51 |
@torch.inference_mode()
|
52 |
def inference_fn(
|
53 |
image,
|
|
|
47 |
model.eval()
|
48 |
conv_mode = "v1"
|
49 |
|
50 |
+
@spaces.GPU(duration=120)
|
51 |
@torch.inference_mode()
|
52 |
def inference_fn(
|
53 |
image,
|