Spaces:
Sleeping
Sleeping
dcavadia commited on
Commit ·
a76fe9c
1
Parent(s): d0847d4
update project structure
Browse files- model_new_new_final.onnx → NFNetL0-0.961.onnx +0 -0
- app.py +2 -2
- dat.json → data.json +0 -0
- model.h5 +0 -3
- model_test2.h5 +0 -3
- modelonnx.onnx +0 -3
- nfnet_model_20epoch_adam_0.962.pkl +0 -3
model_new_new_final.onnx → NFNetL0-0.961.onnx
RENAMED
|
File without changes
|
app.py
CHANGED
|
@@ -12,9 +12,9 @@ import os
|
|
| 12 |
# Model + metadata
|
| 13 |
# ----------------------------
|
| 14 |
ORT_PROVIDERS = ["CPUExecutionProvider"] # add "CUDAExecutionProvider" if available
|
| 15 |
-
ort_session = ort.InferenceSession("
|
| 16 |
|
| 17 |
-
with open("
|
| 18 |
data = json.load(f)
|
| 19 |
CLASSES = list(data) # ordered list of class names
|
| 20 |
|
|
|
|
| 12 |
# Model + metadata
|
| 13 |
# ----------------------------
|
| 14 |
ORT_PROVIDERS = ["CPUExecutionProvider"] # add "CUDAExecutionProvider" if available
|
| 15 |
+
ort_session = ort.InferenceSession("NFNetL0-0.961.onnx", providers=ORT_PROVIDERS)
|
| 16 |
|
| 17 |
+
with open("data.json", "r", encoding="utf-8") as f:
|
| 18 |
data = json.load(f)
|
| 19 |
CLASSES = list(data) # ordered list of class names
|
| 20 |
|
dat.json → data.json
RENAMED
|
File without changes
|
model.h5
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:22a12cc74d4912385b2179aa193cf35cfffcdb4efd9c6c1d3311c8ad79082fcb
|
| 3 |
-
size 131223670
|
|
|
|
|
|
|
|
|
|
|
|
model_test2.h5
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:f95a8f072b45a82dbcbf50f1eaa5ca2f20f98fe2d540687455a5eac3783fe3dd
|
| 3 |
-
size 131287804
|
|
|
|
|
|
|
|
|
|
|
|
modelonnx.onnx
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:15d93dd23fdffedcc74eef5a9644dfc87e55e81df8a160ca0c74f2da400321ff
|
| 3 |
-
size 218626961
|
|
|
|
|
|
|
|
|
|
|
|
nfnet_model_20epoch_adam_0.962.pkl
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:e586c2928820b34b25db8f7b7333da054789038a3e91a1066e624b65de3ad0aa
|
| 3 |
-
size 131902273
|
|
|
|
|
|
|
|
|
|
|
|