Update app.py

app.py

@@ -68,7 +68,7 @@ if file_name is not None:
     nms_prediction = apply_nms(predictions, iou_thresh=0.1)
 
     plot_img_bbox(image, nms_prediction)
-    pred = Image.open("pred.jpg")
+    pred = np.array(Image.open("pred.jpg"))
     col2.image(pred, use_column_width=True)