Spaces:
Runtime error
Runtime error
Sriram Elango
commited on
Commit
•
151f886
1
Parent(s):
7c88712
Update
Browse files- Content/cnn.pt +3 -0
- Content/{best1.pt → train1.pt} +0 -0
- Content/{best2.pt → train2.pt} +0 -0
- app.py +1 -1
Content/cnn.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3babfdd88b4df70b2d8878c15bd17452535fc6b3e6d06e3b14bb1a185da2df07
|
| 3 |
+
size 14329973
|
Content/{best1.pt → train1.pt}
RENAMED
|
File without changes
|
Content/{best2.pt → train2.pt}
RENAMED
|
File without changes
|
app.py
CHANGED
|
@@ -11,7 +11,7 @@ def load_model():
|
|
| 11 |
'''
|
| 12 |
Loading hub model & setting the preferences for the model
|
| 13 |
'''
|
| 14 |
-
model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/
|
| 15 |
model.conf = 0.38
|
| 16 |
model.dnn=True
|
| 17 |
model.agnostic=True
|
|
|
|
| 11 |
'''
|
| 12 |
Loading hub model & setting the preferences for the model
|
| 13 |
'''
|
| 14 |
+
model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/cnn.pt')
|
| 15 |
model.conf = 0.38
|
| 16 |
model.dnn=True
|
| 17 |
model.agnostic=True
|