| """ | |
| Space Gradio — Analyse de sentiment (DistilBERT fine-tuné). | |
| """ | |
| import gradio as gr | |
| from transformers import pipeline | |
| MODEL_ID = "HamzaAS/distilbert-sentiment-demo" | |
| classifier = pipeline("sentiment-analysis", model=MODEL_ID, truncation=True) | |
| def predict(text): | |
| if not text or not text.strip(): | |
| return {} | |
| preds = classifier(text, top_k=None) | |
| return {p["label"]: float(p["score"]) for p in preds} | |
| demo = gr.Interface( | |
| fn=predict, | |
| inputs=gr.Textbox(lines=4, label="Votre texte", placeholder="This movie was fantastic!"), | |
| outputs=gr.Label(num_top_classes=2, label="Sentiment"), | |
| title="🎬 Sentiment Analysis (DistilBERT)" | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() | |