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