Update app.py

app.py

@@ -3,19 +3,35 @@ import pytesseract
 from PIL import Image
 import cv2
 import numpy as np
+import os
+from paddleocr import PaddleOCR, draw_ocr
+from PIL import Image
+import gradio as gr
+import torch
+
 def ocr_image(image):
     """Perform OCR on an image and classify logo."""
     # Convert image to grayscale
-    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-
-    # Perform OCR on the grayscale image
-    text = pytesseract.image_to_string(gray)
-
-    # Classify logo based on presence of space in 'amazonbasics'
+    # gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+
+    # # Perform OCR on the grayscale image
+    # text = pytesseract.image_to_string(gray)
+
+    # # Classify logo based on presence of space in 'amazonbasics'
+  
+    ocr = PaddleOCR(use_angle_cls=True, lang=lang,use_gpu=False)
+    result = ocr.ocr(image, cls=True)[0]
+    # image = Image.open(img_path).convert('RGB')
+    boxes = [line[0] for line in result]
+    txts = [line[1][0] for line in result]
+    text =  '\n'.join(txts)
+    
     if 'amazon basics' in text.lower():
         return 'New Logo'
-    else:
+    elif 'amazonbasics' in text.lower():
         return 'Old Logo'
+    else:
+        return "NA"
 
 # Streamlit interface
 st.title('Logo Classifier')