Update app.py

app.py

@@ -51,8 +51,8 @@ device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cp
 model = torch.load('pickel.pth',map_location=torch.device('cpu'))
 model = model.to(device)
 
-
-def predict(img) -> Tuple[Dict, float]:
+#-> Tuple[Dict, float]
+def predict(img) :
     
     # Start a timer
     start_time = timer()
@@ -71,7 +71,7 @@ def predict(img) -> Tuple[Dict, float]:
 
     nms_prediction = apply_nms(predictions, iou_thresh=0.1)
 
-    plot_img_bbox(img, nms_prediction)
+    plot_img_bbox(np.array(img), nms_prediction)
 
     pred = np.array(Image.open("pred.jpg"))