finalf0 commited on
Commit
eaa3e2c
·
1 Parent(s): 312ea7b

Add spaces GPU

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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: