Aimlab commited on
Commit
061befe
1 Parent(s): 1af7520

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -23,7 +23,7 @@ You can import this model directly from the transformers library:
23
  ```python
24
  >>> from transformers import AutoTokenizer, AutoModelForSequenceClassification
25
  >>> tokenizer = AutoTokenizer.from_pretrained("Aimlab/xlm-roberta-base-finetuned-urdu")
26
- >>> model = AutoModelForSequenceClassification.from_pretrained("Aimlab/xlm-roberta-base-finetuned-urdu")
27
 
28
  ```
29
 
 
23
  ```python
24
  >>> from transformers import AutoTokenizer, AutoModelForSequenceClassification
25
  >>> tokenizer = AutoTokenizer.from_pretrained("Aimlab/xlm-roberta-base-finetuned-urdu")
26
+ >>> model = AutoModelForSequenceClassification.from_pretrained("Aimlab/xlm-roberta-base-finetuned-urdu", id2label = {0: 'negative', 1: 'positive'})
27
 
28
  ```
29