File size: 397 Bytes
ed5c103 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | import gradio as gr
from transformers import pipeline
classifier = pipeline(
"text-classification",
model="distilbert-base-uncased-finetuned-sst-2-english"
)
def predict(text):
result = classifier(text)[0]
return f"{result['label']} ({result['score']:.2f})"
demo = gr.Interface(
fn=predict,
inputs="text",
outputs="text",
title="Simple AI Model"
)
demo.launch() |