gopichandra commited on
Commit
349558e
·
verified ·
1 Parent(s): 076f174

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +5 -4
utils.py CHANGED
@@ -1,7 +1,7 @@
1
  from paddleocr import PaddleOCR
2
  import re
3
 
4
- # Initialize OCR for English Aadhaar/PAN cards
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
- if "MALE" in line.upper():
 
105
  gender = "MALE"
106
  break
107
- elif "FEMALE" in line.upper():
108
  gender = "FEMALE"
109
  break
110
- elif "TRANSGENDER" in line.upper():
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