How to use KapitalK/DummyIMDBModel with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="KapitalK/DummyIMDBModel")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("KapitalK/DummyIMDBModel") model = AutoModelForSequenceClassification.from_pretrained("KapitalK/DummyIMDBModel", device_map="auto")
