Update app.py

app.py

@@ -11,7 +11,7 @@ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 # Load models
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# YOLOv5 for digit detection (Pre-trained model)
+# YOLOv5 for digital meter detection (Pre-trained model)
 yolo_model = YOLO("yolov5s.pt")
 
 # OCR Models
@@ -33,18 +33,18 @@ def convert_to_rgb(image):
         image = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB)
     return image
 
-# Detect Digits Using YOLOv5
-def detect_digits(image):
+# Detect Digital Meter Using YOLOv5
+def detect_meter(image):
     results = yolo_model(image)
-    detected_numbers = []
+    detected_meters = []
 
     for result in results:
         if hasattr(result, "boxes"):  # Ensure correct format
             for box in result.boxes:
                 if box.conf > 0.5:  # Confidence threshold
-                    detected_numbers.append(box.xyxy.tolist())
+                    detected_meters.append(box.xyxy.tolist())
 
-    return detected_numbers
+    return detected_meters
 
 # Extract Text Using EasyOCR
 def extract_text_easyocr(image):
@@ -68,7 +68,7 @@ def extract_weight(text):
 # Full Processing Pipeline
 def process_image(image):
     enhanced = enhance_image(image)
-    detected_digits = detect_digits(image)
+    detected_meters = detect_meter(image)
     text_easyocr = extract_text_easyocr(enhanced)
     text_trocr = extract_text_trocr(enhanced)