Spaces:
Sleeping
Sleeping
Update src/ocr_engine.py
Browse files- src/ocr_engine.py +1 -1
src/ocr_engine.py
CHANGED
@@ -17,6 +17,6 @@ def extract_weight_from_image(pil_img):
|
|
17 |
|
18 |
extracted_text = ''.join(filter(lambda x: x in '0123456789.', ''.join(data['text'])))
|
19 |
confidences = [int(conf) for conf in data['conf'] if conf.isdigit()]
|
20 |
-
avg_conf = sum(confidences)/len(confidences) if confidences else 0
|
21 |
|
22 |
return extracted_text.strip(), avg_conf
|
|
|
17 |
|
18 |
extracted_text = ''.join(filter(lambda x: x in '0123456789.', ''.join(data['text'])))
|
19 |
confidences = [int(conf) for conf in data['conf'] if conf.isdigit()]
|
20 |
+
avg_conf = sum(confidences) / len(confidences) if confidences else 0
|
21 |
|
22 |
return extracted_text.strip(), avg_conf
|