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