SetFit/bbc-news
Viewer โข Updated โข 2.23k โข 1.93k โข 22
How to use AyoubChLin/BERT-Large_BBC_news with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="AyoubChLin/BERT-Large_BBC_news") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("AyoubChLin/BERT-Large_BBC_news")
model = AutoModelForSequenceClassification.from_pretrained("AyoubChLin/BERT-Large_BBC_news")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://api-inference.huggingface.co/models/AyoubChLin/autotrain-bert_bbc_news-48925118418
Or Python API:
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("AyoubChLin/autotrain-bert_bbc_news-48925118418", use_auth_token=True)
tokenizer = AutoTokenizer.from_pretrained("AyoubChLin/autotrain-bert_bbc_news-48925118418", use_auth_token=True)
inputs = tokenizer("I love AutoTrain", return_tensors="pt")
outputs = model(**inputs)