Views
No views yet
bert-base-uncased1from transformers import AutoConfig, AutoTokenizer, AutoModelForSequenceClassification
2
3# Load model and tokenizer
4tokenizer = AutoTokenizer.from_pretrained("real-jiakai/NLP_with_Disaster_Tweets")
5model = AutoModelForSequenceClassification.from_pretrained("real-jiakai/NLP_with_Disaster_Tweets")
6
7# Example usage
8text = "There was a major earthquake in California"
9inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
10outputs = model(**inputs)
11predicted_class = outputs.logits.argmax(-1).item()@misc{NLP_with_Disaster_Tweets,
author = {real-jiakai},
title = {NLP_with_Disaster_Tweets},
year = {2024},
url = {https://huggingface.co/real-jiakai/NLP_with_Disaster_Tweets},
publisher = {Hugging Face}
}