Update app.py

app.py

@@ -211,13 +211,17 @@ def process_image(image, task_prompt, text_input=None, model_id='microsoft/Flore
     elif task_prompt == 'OCR':
         task_prompt = '<OCR>'
         results = run_example(task_prompt, image, model_id=model_id)
-        return results, None
+        # Ensure the OCR results are returned as text output
+        ocr_text = results.get('<OCR>', "")
+        return ocr_text, None  # Return OCR results as text
     elif task_prompt == 'OCR with Region':
         task_prompt = '<OCR_WITH_REGION>'
         results = run_example(task_prompt, image, model_id=model_id)
         output_image = copy.deepcopy(image)
         output_image = draw_ocr_bboxes(output_image, results['<OCR_WITH_REGION>'])
-        return results, output_image
+        # Ensure the OCR text content is extracted for download
+        ocr_text = results.get('<OCR_WITH_REGION>', {}).get('text', "")
+        return ocr_text, output_image
     else:
         return "", None  # Return empty string and None for unknown task prompts