distilbert-disaster-tweet-classification

Disaster Tweet Classification Model

This model is a fine-tuned version of DISTILBERT for classifying tweets as either relating to real disasters or not.

Model Details

• Model Type: DISTILBERT
• Task: Binary Classification (Disaster / Not Disaster)
• Training Data: Disaster Tweets Dataset

Usage

from transformers import AutoModelForSequenceClassification, AutoTokenizer

model = AutoModelForSequenceClassification.from_pretrained("your-username/your-model-name")
tokenizer = AutoTokenizer.from_pretrained("your-username/your-model-name")

# Example usage
text = "There's a wildfire spreading rapidly near the city."
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
prediction = outputs.logits.argmax().item()

print("Disaster" if prediction == 1 else "Not a Disaster")

Replace "your-username/your-model-name" with your Hugging Face username and chosen model name.