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() |