File size: 756 Bytes
09ce0ff
 
 
 
 
 
 
1ae3cd6
 
09ce0ff
 
1ae3cd6
09ce0ff
 
b7f3b84
42e9ef5
09ce0ff
 
 
 
 
 
 
 
 
 
 
 
 
 
 
07a2f7d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import pickle
import gradio as gr

from sklearn.feature_extraction.text import TfidfVectorizer


filename = 'movie_reviews_sentiment_analysis.pkl'
vector = 'vectorizer.pkl'

#Load and test saved model
loaded_model = pickle.load(open(filename, 'rb'))
loaded_vector = pickle.load(open(vector, 'rb'))

def movie_review(review):

    new_review = loaded_vector.fit_transform(review)
    
    result = loaded_model.predict(new_review)
    
    return result

#create input and output objects
#input object1
input = gr.inputs.Textbox(label="Enter Review")

#output object
output = gr.outputs.Textbox(label= "Review Prediction") 

#create interface
gui = gr.Interface(fn=movie_review,
                   inputs=[input],
                   outputs=output).launch()