Spaces:

KaraAgroAI
/

CADI-AI

Runtime error

KaraAgroAI commited on Jun 9, 2023

Commit

5597ce6

1 Parent(s): 8b8ad4e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,12 +10,13 @@ from PIL import Image
 import torch
 import ultralytics
-model = torch.hub.load("ultralytics/yolov5", "custom", path="yolov5_0.65map_exp7_best.pt",
                         force_reload=False)
 model.conf = 0.20  # NMS confidence threshold
-path  = [['img/test-image.jpg']]
 def show_preds_image(im):

 import torch
 import ultralytics
+model = torch.hub.load("ultralytics/yolov5", "custom", path="model/yolov5_0.65map_exp7_best.pt",
                         force_reload=False)
 model.conf = 0.20  # NMS confidence threshold
+# sample test images
+path  = [['sample-images/27.jpg', 'sample-images/82.jpg', 'sample-images/91.jpg']]
 def show_preds_image(im):