Spaces:
Paused
Paused
Commit
•
511497f
1
Parent(s):
f6ccb67
add model
Browse files- .gitattributes +1 -0
- app.py +1 -1
- models/best_model.pth +3 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
models/best_model.pth filter=lfs diff=lfs merge=lfs -text
|
app.py
CHANGED
|
@@ -15,7 +15,7 @@ def demo(img_path):
|
|
| 15 |
# config
|
| 16 |
batch_size = 8
|
| 17 |
crop_size = 256
|
| 18 |
-
model_path = '
|
| 19 |
|
| 20 |
device = torch.device('cuda')
|
| 21 |
|
|
|
|
| 15 |
# config
|
| 16 |
batch_size = 8
|
| 17 |
crop_size = 256
|
| 18 |
+
model_path = './models/best_model.pth'
|
| 19 |
|
| 20 |
device = torch.device('cuda')
|
| 21 |
|
models/best_model.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fbcb717cbf7510ad14005d1875bac97d90c707bd48a9d208c6b5560b1a7ccaae
|
| 3 |
+
size 565718643
|