nlp_sentiment / app.py
amaldevc's picture
Update app.py
97091ef verified
raw
history blame
327 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline(task="sentiment-analysis", model="amaldevc/nlp_sentiment")
def predict(review):
pred = pipe.predict(review)
return pred[0]["label"] + " with score " + pred[0]["score"]
iface = gr.Interface(fn=predict, inputs="text", outputs="text")
iface.launch()