AuditEdge commited on
Commit
a8ec4b1
·
1 Parent(s): 81e13bb

made changes in response

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -158,21 +158,25 @@ def perform_inference(file_paths: Dict[str, str]):
158
  context = aadhar_model
159
  processor = processor_aadhar
160
  name = "aadhar"
 
161
 
162
  if doc_type == "pan_file":
163
  context = pan_model
164
  processor = processor_pan
165
  name = "pan"
 
166
 
167
  if doc_type == "gst_file":
168
  context = gst_model
169
  processor = processor_gst
170
  name = "gst"
 
171
 
172
  if doc_type == "cheque_file":
173
  context = cheque_model
174
  processor = processor_cheque
175
  name = "cheque"
 
176
 
177
 
178
 
@@ -180,7 +184,7 @@ def perform_inference(file_paths: Dict[str, str]):
180
  result = handle(inference_batch, context,processor,name)
181
 
182
  # Store the result
183
- inference_results[doc_type] = result
184
  else:
185
  print(f"Model directory not found for {doc_type}. Skipping.")
186
 
 
158
  context = aadhar_model
159
  processor = processor_aadhar
160
  name = "aadhar"
161
+ attachemnt_num = 3
162
 
163
  if doc_type == "pan_file":
164
  context = pan_model
165
  processor = processor_pan
166
  name = "pan"
167
+ attachemnt_num = 2
168
 
169
  if doc_type == "gst_file":
170
  context = gst_model
171
  processor = processor_gst
172
  name = "gst"
173
+ attachemnt_num = 4
174
 
175
  if doc_type == "cheque_file":
176
  context = cheque_model
177
  processor = processor_cheque
178
  name = "cheque"
179
+ attachemnt_num = 8
180
 
181
 
182
 
 
184
  result = handle(inference_batch, context,processor,name)
185
 
186
  # Store the result
187
+ inference_results["attachment_{}".format(attachemnt_num)] = result
188
  else:
189
  print(f"Model directory not found for {doc_type}. Skipping.")
190