MossyHead commited on
Commit
5b4ebfb
·
1 Parent(s): e2eedda
Files changed (1) hide show
  1. models/p_logreg.py +2 -2
models/p_logreg.py CHANGED
@@ -3,12 +3,12 @@ from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer
3
  import pickle
4
 
5
 
6
- with open('logreg_vec.pkl', 'rb') as f:
7
  vectorizer = pickle.load(f)
8
 
9
 
10
  # Load the model
11
- classifier = load('logreg_model.joblib')
12
 
13
  def predict_tfidf(text):
14
  text_review_vectorized = vectorizer.transform([text])
 
3
  import pickle
4
 
5
 
6
+ with open('models/logreg_vec.pkl', 'rb') as f:
7
  vectorizer = pickle.load(f)
8
 
9
 
10
  # Load the model
11
+ classifier = load('models/logreg_model.joblib')
12
 
13
  def predict_tfidf(text):
14
  text_review_vectorized = vectorizer.transform([text])