File size: 340 Bytes
3f0a7c9 |
1 2 3 4 5 6 7 8 9 10 |
from transformers import pipeline
import gradio as gr
model = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")
def analyze(text):
result = model(text)[0]
return f"Label: {result['label']}, Score: {round(result['score'], 3)}"
gr.Interface(fn=analyze, inputs="text", outputs="text").launch() |