Spaces:
Sleeping
Sleeping
Update utils.py
Browse files
utils.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
from paddleocr import PaddleOCR
|
2 |
import re
|
3 |
|
4 |
-
# Initialize OCR
|
5 |
ocr = PaddleOCR(use_angle_cls=True, lang='en')
|
6 |
|
7 |
def extract_kyc_fields(file_path):
|
@@ -101,13 +101,14 @@ def extract_kyc_fields(file_path):
|
|
101 |
# Gender
|
102 |
gender = "Not found"
|
103 |
for line in lines:
|
104 |
-
|
|
|
105 |
gender = "MALE"
|
106 |
break
|
107 |
-
elif "FEMALE" in
|
108 |
gender = "FEMALE"
|
109 |
break
|
110 |
-
elif "TRANSGENDER" in
|
111 |
gender = "TRANSGENDER"
|
112 |
break
|
113 |
response["gender"] = gender
|
|
|
1 |
from paddleocr import PaddleOCR
|
2 |
import re
|
3 |
|
4 |
+
# Initialize OCR once (English)
|
5 |
ocr = PaddleOCR(use_angle_cls=True, lang='en')
|
6 |
|
7 |
def extract_kyc_fields(file_path):
|
|
|
101 |
# Gender
|
102 |
gender = "Not found"
|
103 |
for line in lines:
|
104 |
+
up = line.upper()
|
105 |
+
if "MALE" in up:
|
106 |
gender = "MALE"
|
107 |
break
|
108 |
+
elif "FEMALE" in up:
|
109 |
gender = "FEMALE"
|
110 |
break
|
111 |
+
elif "TRANSGENDER" in up:
|
112 |
gender = "TRANSGENDER"
|
113 |
break
|
114 |
response["gender"] = gender
|