bondarchukb commited on
Commit
b8eb90a
·
1 Parent(s): 053a766

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -8,7 +8,7 @@ This is best-base-cased model trained on the binary dataset prepared for adverti
8
 
9
  ## Example of classification
10
 
11
- '''python
12
  from transformers import AutoModelForSequenceClassification
13
  from transformers import AutoTokenizer
14
  import numpy as np
@@ -22,7 +22,7 @@ scores = output[0][0].detach().to('cpu').numpy()
22
  scores = softmax(scores)
23
  prediction_class = np.argmax(scores)
24
  print(prediction_class)
25
- '''
26
  Output:
27
  ```
28
  1
 
8
 
9
  ## Example of classification
10
 
11
+ ```python
12
  from transformers import AutoModelForSequenceClassification
13
  from transformers import AutoTokenizer
14
  import numpy as np
 
22
  scores = softmax(scores)
23
  prediction_class = np.argmax(scores)
24
  print(prediction_class)
25
+ ```
26
  Output:
27
  ```
28
  1