Downloads · 30 days
62
37% of all-time downloads
movie-genres-classification
prashantkhandelwal/movie-genres-classification
movie-genres-classification is a text classification model from prashantkhandelwal. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
A BERT-based multi-label classifier that predicts one or more movie genres from a movie’s title and overview. The model supports 19 genres, including Action, Comedy, Drama, Horror, Romance, Science Fiction, and Thriller.
All-time downloads
168
Public
Parameters
109M
1.8 GB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors438 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- bert
- Access
- Public
- Created
- Sep 3, 2026
- Updated
- Sep 11, 2026
- SHA
- 497a8db9
Try a prompt
Base models
Model card
From the Hugging Face model README
Model Description
A BERT-based multi-label classifier that predicts one or more movie genres from a movie’s title and overview. The model supports 19 genres, including Action, Comedy, Drama, Horror, Romance, Science Fiction, and Thriller.
Usage
Load the model with Hugging Face Transformers and provide a movie title or overview. Apply sigmoid to the model outputs and select genres with a probability of at least 0.5.
from transformers import pipeline
classifier = pipeline(
"text-classification",
model="prashantkhandelwal/movie-genres-classification",
top_k=None,
)
result = classifier(
"A crew travels through deep space to stop an alien threat from destroying Earth."
)
genres = [label for label in result[0] if label["score"] >= 0.5]
print(genres)
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- apache-2.0
- Languages
- en
- Created
- Sep 3, 2026
- Updated
- Sep 11, 2026