You can deploy this model and start earning money today!

BilalChafi/
distilbert-disaster-tweet-classification

Name: distilbert-disaster-tweet-classification
Availability: InStock

safetensors

distilbert

Views

No views yet

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

python

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.

BilalChafi/distilbert-disaster-tweet-classification

Disaster Tweet Classification Model

Model Details

Usage

BilalChafi/
distilbert-disaster-tweet-classification