Spaces:

ByteBlaze
/

fake-news-detection-sytem

Runtime error

ByteBlaze commited on Apr 1

Commit

adeed96

•

1 Parent(s): 54db931

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -37,7 +37,7 @@ def classify(text,choice):
       features = vectorizer.transform(corpus).toarray()
       prediction = nb_classifier.predict(features)
   elif(choice == 2):
-      corpus = preprocess_for_bow_and_tfidf
       features = tfidf_vectorizer.transform(corpus).toarray()
       prediction = random_forest.predict(features)
   if(prediction == 1):

       features = vectorizer.transform(corpus).toarray()
       prediction = nb_classifier.predict(features)
   elif(choice == 2):
+      corpus = preprocess_for_bow_and_tfidf(text)
       features = tfidf_vectorizer.transform(corpus).toarray()
       prediction = random_forest.predict(features)
   if(prediction == 1):