facebook
/

regnet-y-032

Image Classification

Inference Endpoints

Model card Files Files and versions Community

arunasank commited on Feb 22, 2023

Commit

45152b6

•

1 Parent(s): 5f29869

Update README.md

Fixed model identifier

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -46,8 +46,8 @@ Here is how to use this model:
 >>> dataset = load_dataset("huggingface/cats-image")
 >>> image = dataset["test"]["image"][0]
->>> feature_extractor = AutoFeatureExtractor.from_pretrained("zuppif/regnet-y-040")
->>> model = RegNetForImageClassification.from_pretrained("zuppif/regnet-y-040")
 >>> inputs = feature_extractor(image, return_tensors="pt")

 >>> dataset = load_dataset("huggingface/cats-image")
 >>> image = dataset["test"]["image"][0]
+>>> feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/regnet-y-032")
+>>> model = RegNetForImageClassification.from_pretrained("facebook/regnet-y-032")
 >>> inputs = feature_extractor(image, return_tensors="pt")