testing / app.py
montoliu's picture
Update app.py
356df8d verified
Raw
History Blame Contribute Delete
326 Bytes
from transformers import pipeline
import gradio as gr
classifier = pipeline("sentiment-analysis")
def predict(prompt):
result = classifier(prompt)
label = result[0]["label"]
score = result[0]["score"]
return label + " " + str(score)
gr.Interface(fn=predict, inputs="text", outputs="text").launch(share=True)