azizbarank commited on
Commit
a872a6b
·
1 Parent(s): 6717c09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ with model_results:
65
  lemmatized_output.append(lemmatizer.lemmatize(word))
66
 
67
  # instantiating count vectorizor
68
- tfidf = joblib.load(open('resources/tfidf_final.pickel', 'rb'))
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)
 
65
  lemmatized_output.append(lemmatizer.lemmatize(word))
66
 
67
  # instantiating count vectorizor
68
+ tfidf = TfidfVectorizer(stop_words= stop_words, ngram_range=(1,2))
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)