Update README.md

README.md

@@ -39,11 +39,17 @@ findings_info =
 """
 Description: PET CT WHOLE BODY
 Radiologist: James
-Findings: Head/Neck: xxx Chest: xxx Abdomen/Pelvis: xxx Extremities/Musculoskeletal: xxx
-Indication: The patient is a 60-year old male with a history of xxx
+Findings:
+Head/Neck: xxx Chest: xxx Abdomen/Pelvis: xxx Extremities/Musculoskeletal: xxx
+Indication:
+The patient is a 60-year old male with a history of xxx
 """
 
-inputs = tokenizer(findings_info.replace('\n', ' '), padding="max_length", truncation=True, max_length=1024, return_tensors="pt")
+inputs = tokenizer(findings_info.replace('\n', ' '),
+                  padding="max_length",
+                  truncation=True,
+                  max_length=1024,
+                  return_tensors="pt")
 input_ids = inputs.input_ids.to("cuda")
 attention_mask = inputs.attention_mask.to("cuda")
 outputs = model.generate(input_ids,
@@ -58,7 +64,9 @@ outputs = model.generate(input_ids,
                         no_repeat_ngram_size=3,
                         early_stopping=True
                         )
-output_str = tokenizer.decode(outputs, skip_special_tokens=True) # get the generated impressions
+# get the generated impressions
+output_str = tokenizer.decode(outputs,
+                              skip_special_tokens=True) 
 ```