khalednabawi11 commited on
Commit
a960399
·
verified ·
1 Parent(s): e25fdf8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -4
README.md CHANGED
@@ -25,8 +25,7 @@ datasets:
25
  You can load the model using the Hugging Face Transformers library:
26
 
27
  ```python
28
- from transformers import AutoTokenizer, AutoConfig
29
- from modeling import ImageCaptioningModel
30
 
31
- config = AutoConfig.from_pretrained("khalednabawi11/Chexnet-MedScan-Report-Gen")
32
- model = ImageCaptioningModel.from_pretrained("khalednabawi11/Chexnet-MedScan-Report-Gen", config=config)
 
25
  You can load the model using the Hugging Face Transformers library:
26
 
27
  ```python
28
+ from transformers import AutoModel, AutoConfig
 
29
 
30
+ config = AutoConfig.from_pretrained("khalednabawi11/Chexnet-MedScan-Report-Gen", trust_remote_code=True)
31
+ model = AutoModel.from_pretrained("khalednabawi11/Chexnet-MedScan-Report-Gen", config=config, trust_remote_code=True)