fix model 3437e3a
moussaKam commited on
How to use moussaKam/frugalscore_small_bert-base_mover-score with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="moussaKam/frugalscore_small_bert-base_mover-score") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("moussaKam/frugalscore_small_bert-base_mover-score")
model = AutoModelForSequenceClassification.from_pretrained("moussaKam/frugalscore_small_bert-base_mover-score", device_map="auto")