| 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](https://huggingface.co/MartinRodrigo/distilbert-sentiment-imdb) | |
| - **GitHub:** [transformer-sentiment-analysis](https://github.com/mrdesautu/ransformer-sentiment-analysis) | |
| ## 💡 Usage | |
| ```python | |
| 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 🚀 |