alexdphan commited on
Commit
1f08cf2
·
1 Parent(s): 2d10e03

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = Pipeline("summarization", model="Mbilal755/Radiology_Bart")
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