Andron00e
/

CLIPForImageClassification-v1

Image Classification

Generated from Trainer

Model card Files Files and versions

Metrics Training metrics Community

Andron00e commited on Dec 13, 2023

Commit

5107e88

·

1 Parent(s): 3a70b14

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -260,7 +260,7 @@ from transformers import CLIPProcessor, AutoModelForImageClassification
 processor = CLIPProcessor.from_pretrained("Andron00e/CLIPForImageClassification-v1")
 model = AutoModelForImageClassification.from_pretrained("Andron00e/CLIPForImageClassification-v1")
-dataset = load_dataset("Andron00e/CIFAR100-custom")
 dataset = dataset["train"].train_test_split(test_size=0.2)
 from datasets import DatasetDict

 processor = CLIPProcessor.from_pretrained("Andron00e/CLIPForImageClassification-v1")
 model = AutoModelForImageClassification.from_pretrained("Andron00e/CLIPForImageClassification-v1")
+dataset = load_dataset("Andron00e/CIFAR10-custom")
 dataset = dataset["train"].train_test_split(test_size=0.2)
 from datasets import DatasetDict