Spaces:
Sleeping
Sleeping
fix(model): correct n_classes
Browse files
shad_mlops_transformers/main.py
CHANGED
@@ -13,7 +13,7 @@ from shad_mlops_transformers.trainer import load_mapper
|
|
13 |
@st.cache_resource
|
14 |
def load_model():
|
15 |
# NOTE hardcoded
|
16 |
-
return DocumentClassifier(n_classes=
|
17 |
|
18 |
|
19 |
mapper = load_mapper()
|
|
|
13 |
@st.cache_resource
|
14 |
def load_model():
|
15 |
# NOTE hardcoded
|
16 |
+
return DocumentClassifier(n_classes=68).from_file()
|
17 |
|
18 |
|
19 |
mapper = load_mapper()
|