---
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 🚀