Create app.py

app.py

@@ -0,0 +1,49 @@
+import cv2
+import numpy as np
+import torch
+import pytesseract
+import easyocr
+import gradio as gr
+from yolov5 import YOLOv5
+import re
+
+# Load YOLOv5 model (pre-trained)
+model = YOLOv5('yolov5s.pt', device='cpu')
+
+# Load EasyOCR
+ocr_reader = easyocr.Reader(['en'])
+
+# Image Preprocessing (Sharpen & Deblur)
+def enhance_image(image):
+    image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    kernel = np.array([[0, -1, 0], [-1, 5, -1], [0, -1, 0]])
+    sharpened = cv2.filter2D(image, -1, kernel)
+    return sharpened
+
+# Detect Digits Using YOLOv5
+def detect_digits(image):
+    results = model(image)
+    digits = [det.xyxy.tolist()[0] for det in results.pred[0] if det.conf > 0.5]
+    return digits
+
+# Extract Text Using OCR
+def extract_text(image):
+    text = pytesseract.image_to_string(image, config='--psm 6')
+    return text
+
+# Extract Weight Using Regex
+def extract_weight(text):
+    matches = re.findall(r'\d+\.\d+', text)  # Extract decimal numbers
+    return matches[0] if matches else "Weight not detected"
+
+# Full Processing Pipeline
+def process_image(image):
+    enhanced = enhance_image(image)
+    digits = detect_digits(image)
+    text = extract_text(enhanced)
+    weight = extract_weight(text)
+    return weight or "No weight detected"
+
+# Gradio Interface
+iface = gr.Interface(fn=process_image, inputs="image", outputs="text")
+iface.launch()