HankyStyle commited on
Commit
4e36a1b
1 Parent(s): c36ae60

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -7
README.md CHANGED
@@ -19,15 +19,20 @@ This model is a fine-tuned version of [bert-base-multilingual-uncased](https://h
19
  ### In Transformers
20
  ```python
21
 
22
- from transformers import AutoTokenizer
23
 
24
- checkpoint = "distilbert-base-uncased-finetuned-sst-2-english"
25
- tokenizer = AutoTokenizer.from_pretrained(checkpoint)
 
 
 
 
 
 
 
 
 
26
 
27
- raw_inputs = [
28
- "I feel happy today!",
29
- ]
30
- inputs = tokenizer(raw_inputs, padding=True, truncation=True, return_tensors="pt")
31
 
32
 
33
  ```
 
19
  ### In Transformers
20
  ```python
21
 
22
+ from transformers import pipeline,AutoTokenizer
23
 
24
+ model_name = "distilbert-base-uncased-finetuned-sst-2-english"
25
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
26
+
27
+ text = "I feel happy today!"
28
+
29
+ inputs = tokenizer(text,return_tensors="pt",padding=True, truncation=True,)
30
+
31
+ {
32
+ 'input_ids': tensor([[ 101, 1045, 2514, 3407, 2651, 999, 102]]),
33
+ 'attention_mask': tensor([[1, 1, 1, 1, 1, 1, 1]])
34
+ }
35
 
 
 
 
 
36
 
37
 
38
  ```