rosemariafontana commited on
Commit
cad3a8c
Β·
verified Β·
1 Parent(s): f9937f7

Changed lists to be min length

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -57,6 +57,15 @@ def parse_ticket_image(image):
57
  true_predictions.append(label)
58
  true_boxes.append(unnormalize_box(token_boxes[idx], width, height))
59
 
 
 
 
 
 
 
 
 
 
60
  data = {
61
  "Field": true_predictions,
62
  "Value": words[1:len(true_predictions)+1]
 
57
  true_predictions.append(label)
58
  true_boxes.append(unnormalize_box(token_boxes[idx], width, height))
59
 
60
+ if len(extracted_fields) == 0:
61
+ print(f"Warning - no fields were extracted!")
62
+ return pd.DataFrame(columns=["Field", "Value"])
63
+
64
+ #Ensure both lists have the same length
65
+ min_length = min(len(fields), len(values))
66
+ fields = fields[:min_length]
67
+ values = values[:min_length]
68
+
69
  data = {
70
  "Field": true_predictions,
71
  "Value": words[1:len(true_predictions)+1]