Update README.md
Browse files
README.md
CHANGED
|
@@ -22,7 +22,7 @@ dataset = load_dataset("huggingface/cats-image")
|
|
| 22 |
image = dataset["test"]["image"][:2]
|
| 23 |
|
| 24 |
processor = AutoImageProcessor.from_pretrained("microsoft/resnet-50")
|
| 25 |
-
model = ResNetModel.from_pretrained("
|
| 26 |
|
| 27 |
inputs = processor(image, return_tensors="pt")
|
| 28 |
|
|
|
|
| 22 |
image = dataset["test"]["image"][:2]
|
| 23 |
|
| 24 |
processor = AutoImageProcessor.from_pretrained("microsoft/resnet-50")
|
| 25 |
+
model = ResNetModel.from_pretrained("aeth0r/cat2vec")
|
| 26 |
|
| 27 |
inputs = processor(image, return_tensors="pt")
|
| 28 |
|