PranomVignesh
commited on
Commit
·
4c6aace
1
Parent(s):
1df9a80
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ imageClassifier = pipeline(task="image-classification",
|
|
10 |
model = torch.hub.load(
|
11 |
'ultralytics/yolov5',
|
12 |
'custom',
|
13 |
-
path='./
|
14 |
device="cpu",
|
15 |
force_reload=True
|
16 |
)
|
|
|
10 |
model = torch.hub.load(
|
11 |
'ultralytics/yolov5',
|
12 |
'custom',
|
13 |
+
path='./best.pt',
|
14 |
device="cpu",
|
15 |
force_reload=True
|
16 |
)
|