Update README.md

README.md

@@ -21,15 +21,19 @@ MediNote-7B is a 7 billion parameters model trained to generate clinical notes f
 - **Code License:** [MIT](https://opensource.org/license/mit/)
 - **Fine-tuned from model:** [Meditron-7B.v1.0](https://huggingface.co/epfl-llm/meditron-7b)
 - **Context length:**  2K tokens
-- **Input:**  Text-only data
-- **Output:**  Model generates text only
-
-### Model Sources
-
+- **Input:**  Patient-doctor conversation transcripts (text)
+- **Output:**  Clinical notes (text)
 - **Repository:** [EPFL-IC-Make-Team/ClinicalNotes](https://github.com/EPFL-IC-Make-Team/ClinicalNotes)
 - **Trainer:** [epflLLM/Megatron-LLM](https://github.com/epfLLM/Megatron-LLM)
 - **Paper:** *[MediNote: Automatic Clinical Notes]()*
 
+<p align="center">
+  <img width=70% src="model_pipeline" alt="Model pipeline" title="Model pipeline">
+</p>
+
+### 
+
+
 ## Uses
 
 ### Direct Use