Spaces:
Sleeping
Sleeping
File size: 714 Bytes
09ce0ff |
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 |
import pickle
import gradio as gr
from sklearn.feature_extraction.text import TfidfVectorizer
tfidf = TfidfVectorizer(stop_words='english')
filename = 'movie_reviews_sentiment_analysis.pkl'
#Load and test saved model
loaded_model = pickle.load(open(filename, 'rb'))
def movie_review(review):
new_review = tfidf.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() |