How to use agarc15/mega-base-wikitext-finetuned-KAGGLE with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="agarc15/mega-base-wikitext-finetuned-KAGGLE")
# Load model directly from transformers import AutoModelForSequenceClassification model = AutoModelForSequenceClassification.from_pretrained("agarc15/mega-base-wikitext-finetuned-KAGGLE", device_map="auto")
SFconvertbot