Spaces:
Sleeping
Sleeping
terapyon
commited on
Commit
·
26879ed
1
Parent(s):
e88a707
change model name from awarefy
Browse files- 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 = "
|
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 |
|