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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -176,6 +176,7 @@ def main():
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)
 
176
  #loaded_vectorizer = dill.load(open('vectorizefile_classification.pickle', 'rb'))
177
  #loaded_vectorizer = pickle.load(open('vectorizefile_classification.pickle', 'rb'))
178
 
179
+ from sklearn.pipeline import Pipeline
180
  pipeline = Pipeline([('count_vect',count_vect),('tfidf_transformer',tfidf_transformer)])
181
  pipeline_test_output = pipeline.fit_transform(class_list)
182
  #pipeline_test_output = loaded_vectorizer.transform(class_list)