PrateekMunjal commited on
Commit
15cf6a4
·
verified ·
1 Parent(s): aaabf97

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -1,7 +1,7 @@
1
  ---
2
  library_name: transformers
3
  base_model:
4
- - facebook/dinov2-with-registers-small
5
  ---
6
 
7
  # Model Card for CXFormer
@@ -48,7 +48,7 @@ The official training splits were used for CheXpert, MIMIC and NIH, and all avai
48
  from transformers import AutoModel, AutoImageProcessor
49
  from PIL import Image
50
 
51
- model_name = 'm42-health/CXformer-small' # or "m42-health/CXformer-base"
52
 
53
  image_processor = AutoImageProcessor.from_pretrained(model_name,trust_remote_code=True)
54
  model = AutoModel.from_pretrained(model_name)
 
1
  ---
2
  library_name: transformers
3
  base_model:
4
+ - facebook/dinov2-with-registers-base
5
  ---
6
 
7
  # Model Card for CXFormer
 
48
  from transformers import AutoModel, AutoImageProcessor
49
  from PIL import Image
50
 
51
+ model_name = "m42-health/CXformer-base"
52
 
53
  image_processor = AutoImageProcessor.from_pretrained(model_name,trust_remote_code=True)
54
  model = AutoModel.from_pretrained(model_name)