Update app.py

app.py

@@ -22,10 +22,10 @@ def box_cxcywh_to_xyxy(x):
 
 
 def rescale_bboxes(out_bbox, size):
-    img_w, img_h = size
-    b = box_cxcywh_to_xyxy(out_bbox)
-    b = b * torch.tensor([img_w, img_h, img_w, img_h], dtype=torch.float32)
-    return b
+    width, height = size
+    boxes = box_cxcywh_to_xyxy(out_bbox)
+    boxes = boxes * torch.tensor([width, height, width, height], dtype=torch.float32)
+    return boxes
 
 
 def outputs_to_objects(outputs, img_size, id2label):
@@ -50,8 +50,8 @@ def fig2img(fig):
     buf = io.BytesIO()
     fig.savefig(buf)
     buf.seek(0)
-    img = Image.open(buf)
-    return img
+    image = Image.open(buf)
+    return image
 
 
 def visualize_detected_tables(img, det_tables):
@@ -117,7 +117,7 @@ def detect_table(image):
     detected_tables = outputs_to_objects(outputs, image.size, id2label)
 
     # visualize
-    fig = visualize_detected_tables(img, detected_tables)
+    fig = visualize_detected_tables(image, detected_tables)
     image = fig2img(fig)
 
     return image