Awlly commited on
Commit
48ac93e
1 Parent(s): a15e210

minor adjustments

Browse files
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 = "/home/owly/ds_bootcamp/phase_2/NLP_processing/misc/logreg_model_v2.joblib"
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):