terapyon commited on
Commit
26879ed
·
1 Parent(s): e88a707

change model name from awarefy

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -16,8 +16,8 @@ from sudachipy import dictionary, tokenizer # type: ignore
16
  # from transformers import AutoModel
17
 
18
  MODELS_PATH = Path(__file__).parent / "saved_model"
19
- model_base_path = MODELS_PATH / "two_class"
20
- MODEL_BASE = "terapyon/awarefy-test-trained-"
21
  topic_model_trained = MODELS_PATH / "topic" / "trained_model.bin"
22
  japanese_selection_path = MODELS_PATH / "stop_words" / "Japanese_selection.txt"
23
 
 
16
  # from transformers import AutoModel
17
 
18
  MODELS_PATH = Path(__file__).parent / "saved_model"
19
+ # model_base_path = MODELS_PATH / "two_class"
20
+ MODEL_BASE = "awarefy/awarefy-two_class-trained-"
21
  topic_model_trained = MODELS_PATH / "topic" / "trained_model.bin"
22
  japanese_selection_path = MODELS_PATH / "stop_words" / "Japanese_selection.txt"
23