Spaces:
Running
Running
Update config.py
Browse files
config.py
CHANGED
|
@@ -3,10 +3,10 @@ config = {
|
|
| 3 |
'tokenizer_path_retrieval': 'imamnurby/bow-tokenizer-uncased',
|
| 4 |
|
| 5 |
# model_path
|
| 6 |
-
'model_path_retrieval': '
|
| 7 |
-
'model_path_generative': '
|
| 8 |
-
'model_path_classifier': '
|
| 9 |
-
'classifier_head_path': '
|
| 10 |
|
| 11 |
# db path
|
| 12 |
'db_metadata_path': 'lib_to_features.csv',
|
|
|
|
| 3 |
'tokenizer_path_retrieval': 'imamnurby/bow-tokenizer-uncased',
|
| 4 |
|
| 5 |
# model_path
|
| 6 |
+
'model_path_retrieval': './library_retriever/retrieval_distillbert',
|
| 7 |
+
'model_path_generative': './pattern_generator',
|
| 8 |
+
'model_path_classifier': './configuration_classifier',
|
| 9 |
+
'classifier_head_path': './configuration_classifier/classifier_head_rf_11nov.pkl',
|
| 10 |
|
| 11 |
# db path
|
| 12 |
'db_metadata_path': 'lib_to_features.csv',
|