Spaces:
Running
on
Zero
Running
on
Zero
aifeifei798
commited on
Upload 4 files
Browse files
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:
|