Gifted030 commited on
Commit
1ae3cd6
1 Parent(s): 9579ac8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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 = tfidf.fit_transform(x)
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