Spaces:
Runtime error
Runtime error
Commit
·
8c04409
1
Parent(s):
1c1adf4
Update app.py
Browse files
app.py
CHANGED
@@ -67,13 +67,13 @@ with model_results:
|
|
67 |
|
68 |
# instantiating count vectorizor
|
69 |
tfidf = TfidfVectorizer(stop_words=stop_words)
|
70 |
-
X_train = joblib.load(open('
|
71 |
X_test = lemmatized_output
|
72 |
X_train_count = tfidf.fit_transform(X_train)
|
73 |
X_test_count = tfidf.transform(X_test)
|
74 |
|
75 |
# loading in model
|
76 |
-
final_model = joblib.load(open('final_bayes.pickle', 'rb'))
|
77 |
|
78 |
# apply model to make predictions
|
79 |
prediction = final_model.predict(X_test_count[0])
|
|
|
67 |
|
68 |
# instantiating count vectorizor
|
69 |
tfidf = TfidfVectorizer(stop_words=stop_words)
|
70 |
+
X_train = joblib.load(open('resources/X_train.pickel', 'rb'))
|
71 |
X_test = lemmatized_output
|
72 |
X_train_count = tfidf.fit_transform(X_train)
|
73 |
X_test_count = tfidf.transform(X_test)
|
74 |
|
75 |
# loading in model
|
76 |
+
final_model = joblib.load(open('resources/final_bayes.pickle', 'rb'))
|
77 |
|
78 |
# apply model to make predictions
|
79 |
prediction = final_model.predict(X_test_count[0])
|