readme

README.md

@@ -30,7 +30,6 @@ library_name: transformers
 Clone repository:
 ```python
 git clone https://huggingface.co/Chantal/RaDialog-interactive-radiology-report-generation
-cd RaDialog-interactive-radiology-report-generation
 ```
 
 Install requirements:
@@ -67,7 +66,7 @@ def load_model_from_huggingface(repo_id):
     model_path = Path(model_path)
 
     tokenizer, model, image_processor, context_len = load_pretrained_model(model_path, model_base='liuhaotian/llava-v1.5-7b',
-                                                                           model_name="llava-v1.5-7b-task-lora_radialog_instruct_llava_biovil_unfrozen_2e-5_5epochs_v5_checkpoint-21000", load_8bit=False, $
+                                                                           model_name="llava-v1.5-7b-task-lora_radialog_instruct_llava_biovil_unfrozen_2e-5_5epochs_v5_checkpoint-21000", load_8bit=False, load_4bit=False)
 
 
     return tokenizer, model, image_processor, context_len
@@ -96,7 +95,7 @@ if __name__ == '__main__':
     findings = ', '.join(findings).lower().strip()
 
     conv = conv_vicuna_v1.copy()
-    REPORT_GEN_PROMPT = f"<image>. Predicted Findings: {findings}. You are to act as a radiologist and write the finding section of a chest x-ray radiology report for this X-ray image and the given predi$
+    REPORT_GEN_PROMPT = f"<image>. Predicted Findings: {findings}. You are to act as a radiologist and write the finding section of a chest x-ray radiology report for this X-ray image and the given predicted findings. Write in the style of a radiologist, write one fluent text without enumeration, be concise and don't provide explanations or reasons."
     print("USER: ", REPORT_GEN_PROMPT)
     conv.append_message("USER", REPORT_GEN_PROMPT)
     conv.append_message("ASSISTANT", None)
@@ -171,6 +170,7 @@ if __name__ == '__main__':
 
     pred = tokenizer.decode(output_ids[0, input_ids.shape[1]:]).strip().replace("</s>", "")
     print("ASSISTANT: ", pred)
+
 ```
 
 ## ✏️ Citation