Spaces:

fair-forward
/

Kara-Agro-Cadi-AI

Sleeping

jonas commited on Jan 10

Commit

58a7962

•

1 Parent(s): ec1be34

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,6 +6,7 @@ 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)

 import torch
 import ultralytics
+model = torch.hub.load('yolov8n.pt')
 model = torch.hub.load("ultralytics/yolov5", "custom", path="yolov5_0.65map_exp7_best.pt",
                         force_reload=False)