Commit
•
f24a3fc
1
Parent(s):
2c39ccc
Update README.md
Browse files
README.md
CHANGED
|
@@ -23,7 +23,7 @@ For more details, check out our [paper](here)
|
|
| 23 |
Finetuning code in the following notebook: [](https://colab.research.google.com/drive/106oPnGhe8B_BCgV6LnJbvVZNv4mCu9Zv?usp=sharing)
|
| 24 |
|
| 25 |
|
| 26 |
-
### Full classification example
|
| 27 |
|
| 28 |
```python
|
| 29 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
|
|
@@ -56,7 +56,7 @@ predicted_class = model.config.id2label[ind.item()]
|
|
| 56 |
```
|
| 57 |
|
| 58 |
|
| 59 |
-
### Pipeline example
|
| 60 |
|
| 61 |
```python
|
| 62 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
|
|
| 23 |
Finetuning code in the following notebook: [](https://colab.research.google.com/drive/106oPnGhe8B_BCgV6LnJbvVZNv4mCu9Zv?usp=sharing)
|
| 24 |
|
| 25 |
|
| 26 |
+
### Full classification example (CPU)
|
| 27 |
|
| 28 |
```python
|
| 29 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
|
|
|
|
| 56 |
```
|
| 57 |
|
| 58 |
|
| 59 |
+
### Pipeline example (CPU & GPU)
|
| 60 |
|
| 61 |
```python
|
| 62 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|