|
|
--- |
|
|
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) |
|
|
1. Create & activate venv: |
|
|
2. Install dependencies (recommended to install torch wheel suitable for your system first): |
|
|
3. Run: |
|
|
|
