Update app.py

app.py

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