Seetha commited on
Commit
31cf0e7
1 Parent(s): 9a1e48e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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
- pipeline_test_output = loaded_vectorizer.transform(class_list)
 
 
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 = []