This model is a fine-tuned version of BERT-base for sentiment analysis tasks.
The model was trained on the Rotten Tomatoes dataset.
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
4model = AutoModelForSequenceClassification.from_pretrained("bert-base-uncased")
5
6input_text = "The movie was fantastic with a gripping storyline!"
7inputs = tokenizer.encode(input_text, return_tensors="pt")
8outputs = model(inputs)
9print(outputs.logits)
The model may generate biased or inappropriate content
due to the nature of the training data.
It is recommended to use the model with caution and apply necessary filters.
This model is licensed under the MIT License.