Update app.py

app.py

@@ -10,10 +10,11 @@ random_forest = joblib.load('./random_forest.pkl')
 
 def classify(text,choice):
   corpus=[text]
-  features = vectorizer.transform(corpus).toarray()
   if(choice == 1):
+      features = vectorizer.transform(corpus).toarray()
       prediction = nb_classifier.predict(features)
   elif(choice == 2):
+      features = tfidf_vectorizer.transform(corpus).toarray()
       prediction = random_forest.predict(features)
   if(prediction == 1):
      return "Fake News"