test / app.py
WYBar's picture
Add application file
3f0a7c9
raw
history blame contribute delete
340 Bytes
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()