Werli commited on
Commit
623807c
·
verified ·
1 Parent(s): d6cb571
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -774,7 +774,7 @@ def process_image(image, task_prompt, text_input=None):
774
  elif task_prompt == 'More Detailed Caption':
775
  task_prompt = '<MORE_DETAILED_CAPTION>'
776
  results = run_example(task_prompt, image)
777
- return results[task_prompt], plot_bbox(image, results['<CAPTION_TO_PHRASE_GROUNDING>'])
778
  elif task_prompt == 'Caption + Grounding':
779
  task_prompt = '<CAPTION>'
780
  results = run_example(task_prompt, image)
 
774
  elif task_prompt == 'More Detailed Caption':
775
  task_prompt = '<MORE_DETAILED_CAPTION>'
776
  results = run_example(task_prompt, image)
777
+ return results, None
778
  elif task_prompt == 'Caption + Grounding':
779
  task_prompt = '<CAPTION>'
780
  results = run_example(task_prompt, image)