Update README.md (#4)
Browse files- Update README.md (1f08cf217edf1e3326556db661d1780c8d4884ca)
Co-authored-by: Alexander Phan <[email protected]>
README.md
CHANGED
|
@@ -96,7 +96,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, Pipeline
|
|
| 96 |
findings = "There is a small lung nodule in the right upper lobe measuring 6 mm. The heart size is normal. No pleural effusion or pneumothorax."
|
| 97 |
|
| 98 |
# Load model & tokenizer
|
| 99 |
-
summarizer =
|
| 100 |
tokenizer = AutoTokenizer.from_pretrained("Mbilal755/Radiology_Bart")
|
| 101 |
|
| 102 |
# Tokenize findings
|
|
|
|
| 96 |
findings = "There is a small lung nodule in the right upper lobe measuring 6 mm. The heart size is normal. No pleural effusion or pneumothorax."
|
| 97 |
|
| 98 |
# Load model & tokenizer
|
| 99 |
+
summarizer = pipeline("summarization", model="Mbilal755/Radiology_Bart")
|
| 100 |
tokenizer = AutoTokenizer.from_pretrained("Mbilal755/Radiology_Bart")
|
| 101 |
|
| 102 |
# Tokenize findings
|