Update README.md
Browse files
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
|
29 |
-
from modeling import ImageCaptioningModel
|
30 |
|
31 |
-
config = AutoConfig.from_pretrained("khalednabawi11/Chexnet-MedScan-Report-Gen")
|
32 |
-
model =
|
|
|
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)
|