minor adjustments
Browse files- main_app.py → app.py +0 -0
- app_models/rubert_MODEL.py +1 -1
main_app.py → app.py
RENAMED
File without changes
|
app_models/rubert_MODEL.py
CHANGED
@@ -12,7 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained(rubert_model_name)
|
|
12 |
model = AutoModel.from_pretrained(rubert_model_name)
|
13 |
|
14 |
# Load Logistic Regression model
|
15 |
-
logreg_model_path = "/
|
16 |
logreg_model = joblib.load(logreg_model_path)
|
17 |
|
18 |
def embed_bert_cls(text, model, tokenizer):
|
|
|
12 |
model = AutoModel.from_pretrained(rubert_model_name)
|
13 |
|
14 |
# Load Logistic Regression model
|
15 |
+
logreg_model_path = "model_data/logreg_model_v2.joblib"
|
16 |
logreg_model = joblib.load(logreg_model_path)
|
17 |
|
18 |
def embed_bert_cls(text, model, tokenizer):
|