Model updated with retrive the detetcted image

app.py

@@ -5,6 +5,7 @@ import easyocr
 import numpy as np
 import re
 from datetime import datetime
+from PIL import Image
 
 # Load YOLO model
 model = YOLO("model/best.pt")
@@ -202,15 +203,20 @@ def extract_receipt(image):
     if not output["time"]:
         output["time"] = now.strftime("%H:%M:%S")
 
-    return output
+    # ---------- Generate YOLO prediction image ----------
+    yolo_img = results.plot()
+    yolo_img = cv2.cvtColor(yolo_img, cv2.COLOR_BGR2RGB)
+    yolo_img_pil = Image.fromarray(yolo_img)
+
+    return output, yolo_img_pil
 
 # ---------- Gradio Interface ----------
 iface = gr.Interface(
     fn=extract_receipt,
     inputs=gr.Image(type="pil"),
-    outputs=gr.JSON(),
+    outputs=[gr.JSON(), gr.Image(type="pil")],
     title="Receipt Extractor",
-    description="Upload a receipt image to extract merchant, date, total, time, and items."
+    description="Upload a receipt image to extract merchant, date, total, time, items, and see YOLO predictions."
 )
 
 iface.launch(share=True)

requirements.txt

@@ -4,4 +4,5 @@ ultralytics==8.3.203
 opencv-python-headless==4.12.0.88
 easyocr==1.7.2
 gradio==5.46.0
-numpy==2.0.2
+numpy==2.0.2
+pillow==11.1.0