Update README.md

README.md

@@ -1,3 +1,48 @@
----
-license: apache-2.0
----
+---
+library_name: peft
+base_model: meta-llama/Llama-2-7b-hf
+license: apache-2.0
+language:
+- en
+---
+## Training:
+
+Check out our githbub: https://github.com/HIRC-SNUBH/Curation_LLM_PathoReport.git
+
+## Inference
+``` python
+from datasets import load_dataset
+from transformers import AutoTokenizer, AutoModelForCausalLM
+from peft import PeftModel
+
+
+base_model = AutoModelForCausalLM.from_pretrained(
+    'meta-llama/Llama-2-7b-chat-hf',
+    trust_remote_code=True,
+    device_map="auto",
+    torch_dtype=torch.float16,   # optional if you have enough VRAM
+)
+tokenizer = AutoTokenizer.from_pretrained('meta-llama/Llama-2-7b-hf')
+
+model = PeftModel.from_pretrained(base_model, 'Lowenzahn/PathoIE-Llama-2-7B')
+model = model.eval()
+```
+
+- PEFT 0.5.0
+
+
+## Citation
+```
+@article{cho2024ie,
+    title={Extracting lung cancer staging descriptors from pathology reports: a generative language model approach},
+    author={Hyeongmin Cho and Sooyoung Yoo and Borham Kim and Sowon Jang and Leonard Sunwoo and Sanghwan Kim and Donghyoung Lee and Seok Kim and Sejin Nam and Jin-Haeng Chung},
+    journal={},
+    volume={},
+    pages={},
+    year={},
+    publisher={},
+    issn={},
+    doi={},
+    url={}
+}
+```