Spaces:

Gifted030
/

movie_reviews_prediction

Sleeping

Gifted030 commited on Feb 21, 2023

Commit

1491a9c

•

1 Parent(s): 07a2f7d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ filename = 'movie_reviews_sentiment_analysis.pkl'
 loaded_model = pickle.load(open(filename, 'rb'))
 def movie_review(review):
-    new_review = tfidf.transform(review)
     result = loaded_model.predict(new_review)

 loaded_model = pickle.load(open(filename, 'rb'))
 def movie_review(review):
+    new_review = tfidf.fit_transform(review)
     result = loaded_model.predict(new_review)