AyoubChLin
/

Bart-MNLI-CNN_news

Zero-Shot Classification

text-classification

text classification

news classification

Inference Endpoints

Model card Files Files and versions Community

AyoubChLin commited on Apr 18, 2023

Commit

6bc7611

·

1 Parent(s): ab0aff8

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -38,9 +38,9 @@ The model can be used for zero-shot text classification tasks on news articles.
 ```python
 from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
-tokenizer = AutoTokenizer.from_pretrained("AyoubChLin/DistilBart_cnn_zeroShot")
-model = AutoModelForSequenceClassification.from_pretrained("AyoubChLin/DistilBart_cnn_zeroShot")
 classifier = pipeline(
     "zero-shot-classification",
     model=model,

 ```python
 from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
+tokenizer = AutoTokenizer.from_pretrained("AyoubChLin/Bart-MNLI-CNN_news")
+model = AutoModelForSequenceClassification.from_pretrained("AyoubChLin/Bart-MNLI-CNN_news")
 classifier = pipeline(
     "zero-shot-classification",
     model=model,