Spaces:
Build error
Build error
add
Browse files- 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])
|