File size: 728 Bytes
dbb32c1 8205a80 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
---
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:
|