Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
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):
|