google-research-datasets/poem_sentiment
Viewer • Updated • 1.1k • 1.38k • 19
How to use ethanrom/a2 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="ethanrom/a2") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("ethanrom/a2")
model = AutoModelForSequenceClassification.from_pretrained("ethanrom/a2")# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("ethanrom/a2")
model = AutoModelForSequenceClassification.from_pretrained("ethanrom/a2")
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="ethanrom/a2")