Update README.md
Browse files
README.md
CHANGED
|
@@ -50,8 +50,8 @@ pip install gliclass
|
|
| 50 |
from gliclass import GLiClassModel, ZeroShotClassificationPipeline
|
| 51 |
from transformers import AutoTokenizer
|
| 52 |
|
| 53 |
-
model = GLiClassModel.from_pretrained("knowledgator/gliclass-instruct-large-
|
| 54 |
-
tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-instruct-large-
|
| 55 |
pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')
|
| 56 |
```
|
| 57 |
|
|
|
|
| 50 |
from gliclass import GLiClassModel, ZeroShotClassificationPipeline
|
| 51 |
from transformers import AutoTokenizer
|
| 52 |
|
| 53 |
+
model = GLiClassModel.from_pretrained("knowledgator/gliclass-instruct-large-v1.0")
|
| 54 |
+
tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-instruct-large-v1.0")
|
| 55 |
pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')
|
| 56 |
```
|
| 57 |
|