Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,17 +7,14 @@ import re
 from datetime import datetime
 import pytz
 
-# Load OCR Model
-ocr = PaddleOCR(use_angle_cls=True, lang='en')  # do not reinstall paddleocr here
+ocr = PaddleOCR(use_angle_cls=True, lang='en')
 
-# Preprocess image: grayscale + threshold
 def preprocess_image(image):
     img = np.array(image.convert("RGB"))
     gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
     _, thresh = cv2.threshold(gray, 150, 255, cv2.THRESH_BINARY)
     return Image.fromarray(thresh)
 
-# Extract weight using regex
 def extract_weight_text(image):
     results = ocr.ocr(np.array(image), cls=True)
     for line in results[0]:
@@ -27,7 +24,6 @@ def extract_weight_text(image):
             return match.group()
     return None
 
-# Streamlit UI
 st.set_page_config(page_title="Auto Weight Logger", layout="centered")
 st.title("📦 Auto Weight Logger (Streamlit)")
 st.write("Upload or capture an image of a digital weight display to extract weight.")
@@ -41,11 +37,9 @@ if input_img is not None:
     image = Image.open(input_img)
     st.image(image, caption="Original Image", use_column_width=True)
 
-    # Preprocess and show preprocessed image
     pre_img = preprocess_image(image)
     st.image(pre_img, caption="Preprocessed Image", use_column_width=True)
 
-    # Detect weight
     weight = extract_weight_text(pre_img)
 
     if weight: