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 |
|