Hugging Face's logo

nvidia
/
MambaVision-T-1K

Image Classification
Transformers
Safetensors
mambavision
image-feature-extraction
custom_code
Model card Files
xet
Community
2
ahatamiz commited on
Commit
e6ba7c2
·
verified ·
1 Parent(s): aa48faa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -5
README.md CHANGED
@@ -37,14 +37,11 @@ huggingface-cli login
37
  The model can be simply used according to:
38
 
39
  ```Python
40
- from transformers import AutoModel
41
- model = AutoModel.from_pretrained("nvidia/MambaVision-T-1K", trust_remote_code=True)
42
  ```
43
 
44
 
45
-
46
-
47
-
48
  ### License:
49
 
50
  [NVIDIA Source Code License-NC](https://huggingface.co/nvidia/MambaVision-T-1K/blob/main/LICENSE)
 
37
  The model can be simply used according to:
38
 
39
  ```Python
40
+ from transformers import AutoModelForImageClassification
41
+ model = AutoModelForImageClassification.from_pretrained("nvidia/MambaVision-T-1K", trust_remote_code=True)
42
  ```
43
 
44
 
 
 
 
45
  ### License:
46
 
47
  [NVIDIA Source Code License-NC](https://huggingface.co/nvidia/MambaVision-T-1K/blob/main/LICENSE)