Update README.md
Browse files
README.md
CHANGED
@@ -31,7 +31,7 @@ You can use this model directly with a pipeline for text classification:
|
|
31 |
```python
|
32 |
>>> from transformers import pipeline
|
33 |
>>> import torch
|
34 |
-
>>> bert_ckpt = "
|
35 |
>>> device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
36 |
>>> pipe = pipeline("text-classification", model=bert_ckpt, device=device)
|
37 |
|
|
|
31 |
```python
|
32 |
>>> from transformers import pipeline
|
33 |
>>> import torch
|
34 |
+
>>> bert_ckpt = "seddiktrk/distilbert-base-uncased-finetuned-clinc"
|
35 |
>>> device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
36 |
>>> pipe = pipeline("text-classification", model=bert_ckpt, device=device)
|
37 |
|