lysandre HF staff dariush-bahrami commited on
Commit
130821f
1 Parent(s): cf73dac

Fix wrong model name in ConvNextV2ForImageClassification (#5)

Browse files

- Fix wrong model name in ConvNextV2ForImageClassification (ee9149df5a5c8c60b9c17d3ef45ef732ac67b956)


Co-authored-by: Dariush Bahrami <[email protected]>

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -44,7 +44,7 @@ dataset = load_dataset("huggingface/cats-image")
44
  image = dataset["test"]["image"][0]
45
 
46
  preprocessor = AutoImageProcessor.from_pretrained("facebook/convnextv2-nano-22k-384")
47
- model = ConvNextV2ForImageClassification.from_pretrained("facebook/convnextv2-huge-22k-384")
48
 
49
  inputs = preprocessor(image, return_tensors="pt")
50
 
 
44
  image = dataset["test"]["image"][0]
45
 
46
  preprocessor = AutoImageProcessor.from_pretrained("facebook/convnextv2-nano-22k-384")
47
+ model = ConvNextV2ForImageClassification.from_pretrained("facebook/convnextv2-nano-22k-384")
48
 
49
  inputs = preprocessor(image, return_tensors="pt")
50