gokaygokay commited on
Commit
beec895
1 Parent(s): d502400

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def fig_to_pil(fig):
42
  buf.seek(0)
43
  return Image.open(buf)
44
 
45
-
46
  def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florence-2-large'):
47
  model = models[model_id]
48
  processor = processors[model_id]
 
42
  buf.seek(0)
43
  return Image.open(buf)
44
 
45
+ @spaces.GPU
46
  def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florence-2-large'):
47
  model = models[model_id]
48
  processor = processors[model_id]