librarian-bots/model_card_dataset_mentions
Viewer โข Updated โข 297 โข 14
How to use librarian-bots/model-card-dataset-mentions with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="librarian-bots/model-card-dataset-mentions") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("librarian-bots/model-card-dataset-mentions")
model = AutoModelForSequenceClassification.from_pretrained("librarian-bots/model-card-dataset-mentions")You can use cURL to access this model:
$ curl -X POST -H "Authorization: Bearer YOUR_API_KEY" -H "Content-Type: application/json" -d '{"inputs": "I love AutoTrain"}' https://huggingface.co/proxy/api-inference.huggingface.co/models/davanstrien/autotrain-dataset-mentions-160223-3522695252
Or Python API:
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("davanstrien/autotrain-dataset-mentions-160223-3522695252", use_auth_token=True)
tokenizer = AutoTokenizer.from_pretrained("davanstrien/autotrain-dataset-mentions-160223-3522695252", use_auth_token=True)
inputs = tokenizer("I love AutoTrain", return_tensors="pt")
outputs = model(**inputs)