dwb2023 commited on
Commit
8746f0d
·
verified ·
1 Parent(s): ef5e1af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -120,7 +120,7 @@ def process_image(image, task_prompt, text_input=None, model_id='dwb2023/florenc
120
  if task_prompt == 'Object Detection':
121
  task_prompt = '<OD>'
122
  parsed_od, generated_od = run_example(task_prompt, image, model_id=model_id)
123
- fig = plot_bbox(image, parsed_answer['<OD>'])
124
  return parsed_od, generated_od, fig_to_pil(fig)
125
  else:
126
  return "", None # Return empty string and None for unknown task prompts
 
120
  if task_prompt == 'Object Detection':
121
  task_prompt = '<OD>'
122
  parsed_od, generated_od = run_example(task_prompt, image, model_id=model_id)
123
+ fig = plot_bbox(image, parsed_od['<OD>'])
124
  return parsed_od, generated_od, fig_to_pil(fig)
125
  else:
126
  return "", None # Return empty string and None for unknown task prompts