J-LAB commited on
Commit
a0a5228
·
verified ·
1 Parent(s): 3d8d1aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -115,7 +115,7 @@ def draw_ocr_bboxes(image, prediction):
115
  fill=color)
116
  return image
117
 
118
- def process_image(image, task_prompt, text_input=None, model_id='microsoft/Florence-2-large'):
119
  image = Image.fromarray(image) # Convert NumPy array to PIL Image
120
  if task_prompt == 'Caption':
121
  task_prompt = '<CAPTION>'
@@ -251,7 +251,7 @@ with gr.Blocks(css=css) as demo:
251
  with gr.Row():
252
  with gr.Column():
253
  input_img = gr.Image(label="Input Picture")
254
- model_selector = gr.Dropdown(choices=list(models.keys()), label="Model", value='microsoft/Florence-2-large')
255
  task_type = gr.Radio(choices=['Single task', 'Cascased task'], label='Task type selector', value='Single task')
256
  task_prompt = gr.Dropdown(choices=single_task_list, label="Task Prompt", value="Caption")
257
  task_type.change(fn=update_task_dropdown, inputs=task_type, outputs=task_prompt)
 
115
  fill=color)
116
  return image
117
 
118
+ def process_image(image, task_prompt, text_input=None, model_id='J-LAB/Florence-Idesire'):
119
  image = Image.fromarray(image) # Convert NumPy array to PIL Image
120
  if task_prompt == 'Caption':
121
  task_prompt = '<CAPTION>'
 
251
  with gr.Row():
252
  with gr.Column():
253
  input_img = gr.Image(label="Input Picture")
254
+ model_selector = gr.Dropdown(choices=list(models.keys()), label="Model", value='J-LAB/Florence-Idesire')
255
  task_type = gr.Radio(choices=['Single task', 'Cascased task'], label='Task type selector', value='Single task')
256
  task_prompt = gr.Dropdown(choices=single_task_list, label="Task Prompt", value="Caption")
257
  task_type.change(fn=update_task_dropdown, inputs=task_type, outputs=task_prompt)