Update app.py

app.py

@@ -27,6 +27,12 @@ def enhance_image(image):
     _, thresholded = cv2.threshold(sharpened, 150, 255, cv2.THRESH_BINARY)
     return thresholded
 
+# Convert Grayscale to RGB (Fix for TrOCR)
+def convert_to_rgb(image):
+    if len(image.shape) == 2:  # Grayscale image
+        image = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB)
+    return image
+
 # Detect Digits Using YOLOv5
 def detect_digits(image):
     results = yolo_model(image)
@@ -45,8 +51,9 @@ def extract_text_easyocr(image):
     text = " ".join(ocr_reader.readtext(image, detail=0))
     return text
 
-# Extract Text Using TrOCR
+# Extract Text Using TrOCR (Fixed)
 def extract_text_trocr(image):
+    image = convert_to_rgb(image)  # Convert grayscale to RGB
     image = Image.fromarray(image)
     pixel_values = trocr_processor(images=image, return_tensors="pt").pixel_values.to(device)
     generated_ids = trocr_model.generate(pixel_values)