aifeifei798 commited on
Commit
21de67a
·
verified ·
1 Parent(s): 6f0c5ee

Upload 4 files

Browse files
Files changed (1) hide show
  1. feifeilib/feifeiflorencebase.py +1 -1
feifeilib/feifeiflorencebase.py CHANGED
@@ -35,7 +35,7 @@ def fig_to_pil(fig):
35
  return Image.open(buf)
36
 
37
  @spaces.GPU
38
- def run_example(task_prompt = "More Detailed Caption", image, text_input=None, model_id='microsoft/Florence-2-base', progress=gr.Progress(track_tqdm=True)):
39
  model = models[model_id]
40
  processor = processors[model_id]
41
  if text_input is None:
 
35
  return Image.open(buf)
36
 
37
  @spaces.GPU
38
+ def run_example(task_prompt = "More Detailed Caption", image = None, text_input = None, model_id='microsoft/Florence-2-base', progress=gr.Progress(track_tqdm=True)):
39
  model = models[model_id]
40
  processor = processors[model_id]
41
  if text_input is None: