Update app.py

app.py

@@ -44,7 +44,11 @@ def visualize_prediction(pil_img, output_dict, threshold=0.5, id2label=None):
         labels = [id2label[x] for x in labels]
 
     plt.figure(figsize=(18, 13))
-    plt.imshow(pil_img)
+    basewidth = 750
+    wpercent = (basewidth/float(pil_img.size[0]))
+    hsize = int((float(pil_img.size[1])*float(wpercent)))
+    img = pil_img.resize((basewidth,hsize), Image.ANTIALIAS)
+    p
     ax = plt.gca()
     colors = COLORS * 100
     for score, (xmin, ymin, xmax, ymax), label, color in zip(scores, boxes, labels, colors):