Update app.py

app.py

@@ -42,13 +42,13 @@ def process_image(image):
         match = re.search(r'(\d{1,3}\.\d{1,3})\s*(kg|g)?', weight)
         if match:
             numeric_value = float(match.group(1))
-            unit = match.group(2) if match.group(2) else extract_unit_from_text(raw_text)
+            unit = match.group(2) if match.group(2) else extract_unit_from_text(raw_text) or "kg"
         else:
             cleaned = re.sub(r"[^\d]", "", weight)
             decimal_fixed = restore_decimal(cleaned)
             try:
                 numeric_value = float(decimal_fixed)
-                unit = extract_unit_from_text(raw_text)
+                unit = extract_unit_from_text(raw_text) or "kg"
             except:
                 return f"❌ Could not extract number | OCR: {weight}", "", image, gr.update(visible=True)