Instructions to use lvwerra/bert-imdb with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use lvwerra/bert-imdb with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="lvwerra/bert-imdb")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("lvwerra/bert-imdb") model = AutoModelForSequenceClassification.from_pretrained("lvwerra/bert-imdb", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
-
Migrate model card from transformers-repo
Read announcement at https://discuss.huggingface.co/t/announcement-all-model-cards-will-be-migrated-to-hf-co-model-repos/2755 Original file history: https://github.com/huggingface/transformers/commits/master/model_cards/lvwerra/bert-imdb/README.md3c84a28