A newer version of the Gradio SDK is available:
6.9.0
metadata
title: Transformer Sentiment Analysis
emoji: 🤖
colorFrom: blue
colorTo: purple
sdk: gradio
sdk_version: 4.44.0
app_file: app.py
pinned: false
license: apache-2.0
tags:
- sentiment-analysis
- transformers
- distilbert
- mlflow
- pytorch
models:
- MartinRodrigo/distilbert-sentiment-imdb
🤖 Transformer Sentiment Analysis
Advanced sentiment analysis using DistilBERT fine-tuned on IMDB dataset with MLflow experiment tracking.
🎯 Model Performance
- Accuracy: 80% on IMDB test set
- F1 Score: 0.7981
- Model: DistilBERT (66M parameters)
- Speed: ~100ms per prediction
🚀 Features
- Real-time sentiment analysis
- Batch text processing
- Confidence scores and probabilities
- Interactive visualizations
🔗 Links
- Model Repository: MartinRodrigo/distilbert-sentiment-imdb
- GitHub: transformer-sentiment-analysis
💡 Usage
from transformers import pipeline
classifier = pipeline("sentiment-analysis",
model="MartinRodrigo/distilbert-sentiment-imdb")
result = classifier("I love this movie!")
print(result)
Built with Transformers, MLflow, and Gradio 🚀