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)
|