How to use GautamR/disaster_classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="GautamR/disaster_classifier")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("GautamR/disaster_classifier") model = AutoModelForSequenceClassification.from_pretrained("GautamR/disaster_classifier")
Label to disaster mapping : {'Coronavirus': 0, 'Earthquake': 1, 'Extreme Heat': 2, 'Fire': 3, 'Flood': 4, 'Terrorist Attack': 5, 'Thunder': 6}
Files info
