English
grantg123 commited on
Commit
55e7e6f
1 Parent(s): d0332ab

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -3
README.md CHANGED
@@ -10,12 +10,21 @@ license: apache-2.0
10
  language:
11
  - en
12
  !pip install simpletransformers
 
13
  !pip freeze | grep simpletransformers
14
- model_name = 'bert-base-uncased' # Example: base BERT model with uncased vocabulary
15
- tokenizer = BertTokenizer.from_pretrained(model_name)
16
- model = BertModel.from_pretrained(model_name)
 
 
 
 
17
  input_text = "You are hurting my feelings. You must be mean."
 
18
  tokenized_input = tokenizer(input_text, return_tensors='pt')
 
19
  outputs = model(**tokenized_input)
 
20
  hidden_states = outputs.last_hidden_state
 
21
  pooled_output = outputs.pooler_output
 
10
  language:
11
  - en
12
  !pip install simpletransformers
13
+
14
  !pip freeze | grep simpletransformers
15
+
16
+ model_name = 'bert-base-uncased'
17
+
18
+ tokenizer = BertTokenizer.from_pretrained
19
+
20
+ model = BertModel.from_pretrained
21
+
22
  input_text = "You are hurting my feelings. You must be mean."
23
+
24
  tokenized_input = tokenizer(input_text, return_tensors='pt')
25
+
26
  outputs = model(**tokenized_input)
27
+
28
  hidden_states = outputs.last_hidden_state
29
+
30
  pooled_output = outputs.pooler_output