Update app.py

app.py

@@ -4,6 +4,8 @@ from PIL import Image, ImageEnhance, ImageOps
 import torch
 import re
 import easyocr
+from io import BytesIO
+import numpy as np
 
 # Initialize EasyOCR Reader
 reader = easyocr.Reader(['en'])
@@ -31,12 +33,17 @@ def extract_details(text):
 
 def extract_meter_reading(image):
     try:
-        # Enhance image
+        # Convert NumPy array to PIL Image
         image = Image.fromarray(image)
         enhanced_image = enhance_image(image)
 
+        # Convert PIL image to bytes for EasyOCR
+        img_byte_arr = BytesIO()
+        enhanced_image.save(img_byte_arr, format='PNG')
+        img_byte_arr = img_byte_arr.getvalue()
+
         # Use EasyOCR to get better results
-        result = reader.readtext(enhanced_image)
+        result = reader.readtext(img_byte_arr)
         text = " ".join([item[1] for item in result])
 
         # Extract details from the OCR text