Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,16 +3,17 @@ import gradio as gr
|
|
3 |
|
4 |
from sklearn.feature_extraction.text import TfidfVectorizer
|
5 |
|
6 |
-
tfidf = TfidfVectorizer(stop_words='english')
|
7 |
|
8 |
filename = 'movie_reviews_sentiment_analysis.pkl'
|
|
|
|
|
9 |
#Load and test saved model
|
10 |
loaded_model = pickle.load(open(filename, 'rb'))
|
|
|
11 |
|
12 |
def movie_review(review):
|
13 |
-
x= tfidf.fit_transform(review)
|
14 |
|
15 |
-
new_review =
|
16 |
|
17 |
result = loaded_model.predict(new_review)
|
18 |
|
|
|
3 |
|
4 |
from sklearn.feature_extraction.text import TfidfVectorizer
|
5 |
|
|
|
6 |
|
7 |
filename = 'movie_reviews_sentiment_analysis.pkl'
|
8 |
+
vector = 'vectorizer.pkl'
|
9 |
+
|
10 |
#Load and test saved model
|
11 |
loaded_model = pickle.load(open(filename, 'rb'))
|
12 |
+
loaded_vector = pickle.load(open(vector, 'rb'))
|
13 |
|
14 |
def movie_review(review):
|
|
|
15 |
|
16 |
+
new_review = loaded_vector.fit_transform([review])
|
17 |
|
18 |
result = loaded_model.predict(new_review)
|
19 |
|