alekeik1 commited on
Commit
98b2c27
1 Parent(s): 2b76117

fix(model): correct n_classes

Browse files
Files changed (1) hide show
  1. shad_mlops_transformers/main.py +1 -1
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=63).from_file()
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()