metadata
title: RoBERTa IMDb Sentiment Analyzer — Pro
emoji: 🤖
colorFrom: blue
colorTo: purple
sdk: docker
sdk_version: latest
app_file: app.py
pinned: true
RoBERTa IMDb Sentiment Analyzer — Pro (Docker-ready)
This project provides a professional-grade sentiment analysis API & UI using a RoBERTa model fine-tuned on the IMDb dataset.
Features
- Model:
textattack/roberta-base-imdb(RoBERTa fine-tuned on IMDb) - Flask web UI + /api/predict JSON endpoint
- Dockerfile for Hugging Face Spaces (Docker SDK)
- Production-ready (port binding, HF cache fix)
Local Testing (Optional)
- Create & activate venv:
- Install dependencies (recommended to install torch wheel suitable for your system first):
- Run: