Hugging Face's logo

facebook
/
regnet-y-032

Image Classification
Transformers
PyTorch
TensorFlow
regnet
vision
Model card Files
xet
 Community
3

Update README.md

#2
by arunasank - opened
base: refs/heads/main
←
from: refs/pr/2
Discussion Files changed
+2
-2
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -46,8 +46,8 @@ Here is how to use this model:
46
  >>> dataset = load_dataset("huggingface/cats-image")
47
  >>> image = dataset["test"]["image"][0]
48
 
49
- >>> feature_extractor = AutoFeatureExtractor.from_pretrained("zuppif/regnet-y-040")
50
- >>> model = RegNetForImageClassification.from_pretrained("zuppif/regnet-y-040")
51
 
52
  >>> inputs = feature_extractor(image, return_tensors="pt")
53
 
 
46
  >>> dataset = load_dataset("huggingface/cats-image")
47
  >>> image = dataset["test"]["image"][0]
48
 
49
+ >>> feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/regnet-y-032")
50
+ >>> model = RegNetForImageClassification.from_pretrained("facebook/regnet-y-032")
51
 
52
  >>> inputs = feature_extractor(image, return_tensors="pt")
53