Update app.py

app.py

@@ -72,7 +72,7 @@ def process_image(image):
         predicted_label = id2label[predicted_labels[idx]]
         box_width = np.array(box)[2] - np.array(box)[0]
         box_height = np.array(box)[3] - np.array(box)[1]
-        normalized_box = unnormalize_box(box, width, height)
+        normalized_box = [int(coord) for coord in unnormalize_box(box, width, height)]
         extracted_content[predicted_label] = image.crop(normalized_box).copy()
 
     # Draw predictions over the image
@@ -82,7 +82,7 @@ def process_image(image):
         predicted_label = iob_to_label(id2label[prediction])
         box_width = np.array(box)[2] - np.array(box)[0]
         box_height = np.array(box)[3] - np.array(box)[1]
-        normalized_box = unnormalize_box(box, width, height)
+        normalized_box = [int(coord) for coord in unnormalize_box(box, width, height)]
         draw.rectangle(normalized_box, outline=label2color[predicted_label])
         draw.text((normalized_box[0] + 10, normalized_box[1] - 10), text=predicted_label, fill=label2color[predicted_label], font=font)