kluvin's picture
Upload folder using huggingface_hub
73bcf15 verified
metadata
title: Tweet Sentiment Classifier
emoji: 🐦
colorFrom: blue
colorTo: green
sdk: docker
app_port: 7860
pinned: false

Tweet Sentiment Classifier

Compare 5 different sentiment analysis models side-by-side on tweet text:

  • BERTweet (Transformer) - Fine-tuned from vinai/bertweet-base
  • Decision Tree (TF-IDF)
  • Random Forest (TF-IDF)
  • Logistic Regression (TF-IDF)
  • Linear SVM (TF-IDF)

How it works

Type or paste a tweet and click "Analyze Sentiment" to see predictions from all 5 models simultaneously.

First run

Models download and train on first request (~2-3 minutes). Subsequent requests are instant thanks to caching.

Tech Stack

  • Frontend: HTMX for reactive updates
  • Backend: Flask
  • Models: HuggingFace Transformers (BERTweet) + scikit-learn
  • Data: SetFit/tweet_sentiment_extraction dataset

Built as a class project for DAT158 Machine Learning.