Update app.py

app.py

@@ -72,8 +72,8 @@ def process_image(image):
     extracted_content = {}
     for idx, box in enumerate(encoding.bbox.tolist()):
         predicted_label = id2label[predicted_labels[idx]]
-        box_width = box[2] - box[0]
-        box_height = box[3] - box[1]
+        box_width = np.array(box)[2] - np.array(box)[0]
+        box_height = np.array(box)[3] - np.array(box)[1]
         normalized_box = [
             box[0] * width / 1000,
             box[1] * height / 1000,
@@ -87,8 +87,8 @@ def process_image(image):
     font = ImageFont.load_default()
     for prediction, box in zip(predicted_labels, encoding.bbox.tolist()):
         predicted_label = iob_to_label(id2label[prediction])
-        box_width = box[2] - box[0]
-        box_height = box[3] - box[1]
+        box_width = np.array(box)[2] - np.array(box)[0]
+        box_height = np.array(box)[3] - np.array(box)[1]
         normalized_box = [
             box[0] * width / 1000,
             box[1] * height / 1000,
@@ -120,3 +120,4 @@ iface = gr.Interface(fn=process_image,
                      enable_queue=True)
 
 iface.launch(inline=False, share=False, debug=False)
+