gokaygokay commited on
Commit
dfe94f3
1 Parent(s): 5aa42a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -90,6 +90,7 @@ def draw_ocr_bboxes(image, prediction):
90
  return image
91
 
92
  def process_image(image, task_prompt, text_input=None):
 
93
  if task_prompt == '<CAPTION>':
94
  result = run_example(task_prompt, image)
95
  return result
 
90
  return image
91
 
92
  def process_image(image, task_prompt, text_input=None):
93
+ image = Image.fromarray(image)
94
  if task_prompt == '<CAPTION>':
95
  result = run_example(task_prompt, image)
96
  return result