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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -171,7 +171,8 @@ def main():
171
  filename = 'Checkpoint-classification.sav'
172
  count_vect = CountVectorizer(ngram_range=[1,3])
173
  tfidf_transformer=TfidfTransformer()
174
- loaded_model = pickle.load(open(filename, 'rb'))
 
175
  loaded_vectorizer = dill.load(open('vectorizefile_classification.pickle', 'rb'))
176
  #loaded_vectorizer = pickle.load(open('vectorizefile_classification.pickle', 'rb'))
177
 
 
171
  filename = 'Checkpoint-classification.sav'
172
  count_vect = CountVectorizer(ngram_range=[1,3])
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