Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -173,10 +173,12 @@ def main():
|
|
173 |
tfidf_transformer=TfidfTransformer()
|
174 |
#loaded_model = pickle.load(open(filename, 'rb'))
|
175 |
loaded_model = dill.load(open(filename, 'rb'))
|
176 |
-
loaded_vectorizer = dill.load(open('vectorizefile_classification.pickle', 'rb'))
|
177 |
#loaded_vectorizer = pickle.load(open('vectorizefile_classification.pickle', 'rb'))
|
178 |
|
179 |
-
|
|
|
|
|
180 |
predicted = loaded_model.predict(pipeline_test_output)
|
181 |
pred1 = predicted
|
182 |
level0 = []
|
|
|
173 |
tfidf_transformer=TfidfTransformer()
|
174 |
#loaded_model = pickle.load(open(filename, 'rb'))
|
175 |
loaded_model = dill.load(open(filename, 'rb'))
|
176 |
+
#loaded_vectorizer = dill.load(open('vectorizefile_classification.pickle', 'rb'))
|
177 |
#loaded_vectorizer = pickle.load(open('vectorizefile_classification.pickle', 'rb'))
|
178 |
|
179 |
+
pipeline = Pipeline([('count_vect',count_vect),('tfidf_transformer',tfidf_transformer)])
|
180 |
+
pipeline_test_output = pipeline.fit_transform(class_list)
|
181 |
+
#pipeline_test_output = loaded_vectorizer.transform(class_list)
|
182 |
predicted = loaded_model.predict(pipeline_test_output)
|
183 |
pred1 = predicted
|
184 |
level0 = []
|