Andron00e commited on
Commit
5107e88
1 Parent(s): 3a70b14

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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/CIFAR100-custom")
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