Update app.py
Browse files
app.py
CHANGED
|
@@ -13,8 +13,8 @@ def load_model():
|
|
| 13 |
'''
|
| 14 |
model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
|
| 15 |
model.conf = 0.38
|
| 16 |
-
model.dnn=
|
| 17 |
-
model.
|
| 18 |
return model
|
| 19 |
|
| 20 |
model=load_model()
|
|
|
|
| 13 |
'''
|
| 14 |
model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
|
| 15 |
model.conf = 0.38
|
| 16 |
+
model.dnn=True
|
| 17 |
+
model.agnostic=True
|
| 18 |
return model
|
| 19 |
|
| 20 |
model=load_model()
|