ananthvk's picture
add code + models
e1fbc2d
raw
history blame contribute delete
565 Bytes
from transformers import pipeline
import gradio as gr
classifier = pipeline("text-classification", model="model")
def detect_sentiment(text):
pred = classifier(text)
if len(pred) > 0:
pred = pred[0]
if pred['label'] == 'LABEL_1':
pred['label'] = "positive"
else:
pred['label'] = "negative"
return {pred['label'] : pred['score']}
return "Internal error"
review = gr.Textbox()
label = gr.Label()
inter = gr.Interface(fn=detect_sentiment, inputs=review, outputs=label)
inter.launch(inline=False)