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
|