Update README.md
Browse files
README.md
CHANGED
|
@@ -50,10 +50,6 @@ tensor([[0.9756, 0.0244]], grad_fn=<SoftmaxBackward0>)
|
|
| 50 |
```python
|
| 51 |
from transformers import AutoModel, AutoProcessor, pipeline
|
| 52 |
|
| 53 |
-
pipe = pipeline("zero-shot-image-classification", model="Bingsu/vitB32_bert_ko_small_clip")
|
| 54 |
-
|
| 55 |
-
# or
|
| 56 |
-
|
| 57 |
model = AutoModel.from_pretrained("Bingsu/vitB32_bert_ko_small_clip")
|
| 58 |
processor = AutoProcessor.from_pretrained("Bingsu/vitB32_bert_ko_small_clip")
|
| 59 |
pipe = pipeline("zero-shot-image-classification", model=model, feature_extractor=processor.feature_extractor, tokenizer=processor.tokenizer)
|
|
|
|
| 50 |
```python
|
| 51 |
from transformers import AutoModel, AutoProcessor, pipeline
|
| 52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
model = AutoModel.from_pretrained("Bingsu/vitB32_bert_ko_small_clip")
|
| 54 |
processor = AutoProcessor.from_pretrained("Bingsu/vitB32_bert_ko_small_clip")
|
| 55 |
pipe = pipeline("zero-shot-image-classification", model=model, feature_extractor=processor.feature_extractor, tokenizer=processor.tokenizer)
|