Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|