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