# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("tensorboy/disaster_classifier_1")
model = AutoModelForSequenceClassification.from_pretrained("tensorboy/disaster_classifier_1")Quick Links
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
label_mapping = {'Coronavirus': 0, 'Earthquake': 1, 'Extreme Heat': 2, 'Fire': 3, 'Flood': 4, 'Terrorist Attack': 5, 'Thunder': 6, "Disaster": 7, "Sunstroke": 8}
- Downloads last month
- 11
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="tensorboy/disaster_classifier_1")