# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("sethkimmel3/distilbert-aviation-classification")
model = AutoModelForSequenceClassification.from_pretrained("sethkimmel3/distilbert-aviation-classification", device_map="auto")Quick Links
This is a simple binary text classification model, fine-tuned on distilbert to determine whether or not a block of text is or is not related to aviation. It was fine-tuned using the ASRS and cc_news datasets. You can view the notebook used for fine-tuning here.
- Downloads last month
- 3
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="sethkimmel3/distilbert-aviation-classification")