Spaces:
Runtime error
Runtime error
Add spaces GPU
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
#!/usr/bin/env python
|
2 |
# encoding: utf-8
|
|
|
3 |
import gradio as gr
|
4 |
from PIL import Image
|
5 |
import traceback
|
@@ -133,7 +134,7 @@ def create_component(params, comp='Slider'):
|
|
133 |
interactive=True
|
134 |
)
|
135 |
|
136 |
-
|
137 |
def chat(img, msgs, ctx, params=None, vision_hidden_states=None):
|
138 |
default_params = {"num_beams":3, "repetition_penalty": 1.2, "max_new_tokens": 1024}
|
139 |
if params is None:
|
|
|
1 |
#!/usr/bin/env python
|
2 |
# encoding: utf-8
|
3 |
+
import spaces
|
4 |
import gradio as gr
|
5 |
from PIL import Image
|
6 |
import traceback
|
|
|
134 |
interactive=True
|
135 |
)
|
136 |
|
137 |
+
@spaces.GPU(duration=120)
|
138 |
def chat(img, msgs, ctx, params=None, vision_hidden_states=None):
|
139 |
default_params = {"num_beams":3, "repetition_penalty": 1.2, "max_new_tokens": 1024}
|
140 |
if params is None:
|