Update app.py

app.py

@@ -1,26 +1,23 @@
 import gradio as gr
 from PIL import Image
 import easyocr
-import cv2
 import numpy as np
-import paddleocr
+from paddleocr import PaddleOCR
 import pytesseract
 
 # Initialize OCR models
 easyocr_reader = easyocr.Reader(['en'])
-paddleocr_reader = paddleocr.OCR()
-# Tesseract requires only function call
+paddleocr_reader = PaddleOCR()
+# Tesseract uses pytesseract directly
 
-# Simulation of metrics, as per user scenario
 def get_metrics(model):
-    # Simulated fixed scores according to user needs
-    base = 0.95
+    # Simulated fixed scores according to user specs
     if model == "EasyOCR":
-        return {"Accuracy": 0.95, "Precision": 0.94, "Pipeline": "Easy Integration (90%)"}
+        return {"Accuracy": "95%", "Precision": "94%", "Pipeline": "Easy Integration (90%)"}
     elif model == "PaddleOCR":
-        return {"Accuracy": 0.90, "Precision": 0.89, "Pipeline": "Moderate Integration (85%)"}
+        return {"Accuracy": "90%", "Precision": "89%", "Pipeline": "Moderate Integration (85%)"}
     elif model == "Tesseract":
-        return {"Accuracy": 0.85, "Precision": 0.83, "Pipeline": "Manual Integration (75%)"}
+        return {"Accuracy": "85%", "Precision": "83%", "Pipeline": "Manual Integration (75%)"}
     else:
         return {}
 
@@ -31,11 +28,11 @@ def ocr_all(img, ground_truth):
     metrics_easy = get_metrics("EasyOCR")
     
     # PaddleOCR
-    result_paddle, _ = paddleocr_reader.ocr(img, cls=True)
+    result_paddle = paddleocr_reader.ocr(np.array(img), cls=True)
     try:
         text_paddle = " ".join([line[1][0] for line in result_paddle])
     except Exception:
-        text_paddle = "Error"
+        text_paddle = "Error during PaddleOCR"
     metrics_paddle = get_metrics("PaddleOCR")
     
     # Tesseract
@@ -61,4 +58,4 @@ with gr.Blocks() as demo:
 
     btn.click(ocr_all, inputs=[img_in, txt_in], outputs=[out_easy, out_paddle, out_tess])
 
-demo.launch()
+demo.launch()