Spaces:
Runtime error
Runtime error
Changed lists to be min length
Browse files
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]
|