ByteBlaze commited on
Commit
54db931
1 Parent(s): 9725d40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ random_forest = joblib.load('./random_forest.pkl')
33
  def classify(text,choice):
34
  corpus=[text]
35
  if(choice == 1):
36
- corpus = preprocess_for_bow(text)
37
  features = vectorizer.transform(corpus).toarray()
38
  prediction = nb_classifier.predict(features)
39
  elif(choice == 2):
 
33
  def classify(text,choice):
34
  corpus=[text]
35
  if(choice == 1):
36
+ corpus = preprocess_for_bow_and_tfidf(text)
37
  features = vectorizer.transform(corpus).toarray()
38
  prediction = nb_classifier.predict(features)
39
  elif(choice == 2):