Update README.md
Browse files
README.md
CHANGED
@@ -260,7 +260,7 @@ from transformers import CLIPProcessor, AutoModelForImageClassification
|
|
260 |
processor = CLIPProcessor.from_pretrained("Andron00e/CLIPForImageClassification-v1")
|
261 |
model = AutoModelForImageClassification.from_pretrained("Andron00e/CLIPForImageClassification-v1")
|
262 |
|
263 |
-
dataset = load_dataset("Andron00e/
|
264 |
dataset = dataset["train"].train_test_split(test_size=0.2)
|
265 |
from datasets import DatasetDict
|
266 |
|
|
|
260 |
processor = CLIPProcessor.from_pretrained("Andron00e/CLIPForImageClassification-v1")
|
261 |
model = AutoModelForImageClassification.from_pretrained("Andron00e/CLIPForImageClassification-v1")
|
262 |
|
263 |
+
dataset = load_dataset("Andron00e/CIFAR10-custom")
|
264 |
dataset = dataset["train"].train_test_split(test_size=0.2)
|
265 |
from datasets import DatasetDict
|
266 |
|