How to use stromdotcom/imdb-finetuned-distilbert-base-uncased with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="stromdotcom/imdb-finetuned-distilbert-base-uncased")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("stromdotcom/imdb-finetuned-distilbert-base-uncased") model = AutoModelForSequenceClassification.from_pretrained("stromdotcom/imdb-finetuned-distilbert-base-uncased", device_map="auto")
A test model repo, distilbert-base-uncased trained for sentiment analysis using the imdb reviews dataset.
