Add library name and link to paper page

README.md

@@ -1,10 +1,11 @@
 ---
-license: apache-2.0
+base_model:
+- FreedomIntelligence/HuatuoGPT-o1-8B
 language:
 - en
-base_model:
-- FreedomIntelligence/HuatuoGPT-o1-8B 
+license: apache-2.0
 pipeline_tag: text-generation
+library_name: transformers
 tags:
 - medical
 - medical-reasoning
@@ -20,7 +21,7 @@ tags:
 </div>
 
 <div align="center">
-<a href="https://github.com/TianYin123/Med-REFL" target="_blank">GitHub</a> | <a href="https://arxiv.org/abs/2504.12334" target="_blank">Paper</a>
+<a href="https://github.com/TianYin123/Med-REFL" target="_blank">GitHub</a> | <a href="https://huggingface.co/papers/2506.13793" target="_blank">Paper</a>
 </div>
 
 # <span>Introduction</span>
@@ -38,7 +39,7 @@ The Med-REFL LoRA weights can be applied to the following base models to enhance
 | :--- | :--- | :--- |
 | **Med-REFL for Llama-3.1-8B** | Llama-3.1-8B | [HF Link](https://huggingface.co/HANI-LAB/Med-REFL-Llama-3.1-8B-lora) |
 | **Med-REFL for Qwen2.5-7B** | Qwen2.5-7B | [HF Link](https://huggingface.co/HANI-LAB/Med-REFL-Qwen2.5-7B-lora) |
-| **Med-REFL for Huatuo-o1-8B** | Huatuo-o1-8b | [HF Link](https://huggingface.co/HANI-LAB/Med-REFL-Huatuo-o1-8B-lora) |
+| **Med-REFL for Huatuo-o1-8B** | Huatuo-o1-8B | [HF Link](https://huggingface.co/HANI-LAB/Med-REFL-Huatuo-o1-8B-lora) |
 | **Med-REFL for MedReason-8B**| MedReason-8B | [HF Link](https://huggingface.co/HANI-LAB/Med-REFL-MedReason-8B-lora) |
 
 # <span>Usage</span>
@@ -67,8 +68,17 @@ base_model = AutoModelForCausalLM.from_pretrained(
 model = PeftModel.from_pretrained(base_model, lora_path)
 
 # Prepare the prompt
-system_prompt = '''You are a helpful medical expert specializing in USMLE exam questions, and your task is to answer a multi-choice medical question. Please first think step-by-step and then choose the answer from the provided options. Your responses will be used for research purposes only, so please have a definite answer.\nProvide your response in the following JSON format:\n{"reason": "Step-by-step explanation of your thought process","answer": "Chosen answer from the given options"}\n'''
-user_prompt = "A 67-year-old man with transitional cell carcinoma of the bladder comes to the physician because of a 2-day history of ringing sensation in his ear. He received this first course of neoadjuvant chemotherapy 1 week ago. Pure tone audiometry shows a sensorineural hearing loss of 45 dB. The expected beneficial effect of the drug that caused this patient's symptoms is most likely due to which of the following actions?\nOptions:\nA: Inhibition of thymidine synthesis\nB: Inhibition of proteasome\nC: Hyperstabilization of microtubules\nD: Generation of free radicals\nE: Cross-linking of DNA"
+system_prompt = '''You are a helpful medical expert specializing in USMLE exam questions, and your task is to answer a multi-choice medical question. Please first think step-by-step and then choose the answer from the provided options. Your responses will be used for research purposes only, so please have a definite answer.
+Provide your response in the following JSON format:
+{"reason": "Step-by-step explanation of your thought process","answer": "Chosen answer from the given options"}
+'''
+user_prompt = "A 67-year-old man with transitional cell carcinoma of the bladder comes to the physician because of a 2-day history of ringing sensation in his ear. He received this first course of neoadjuvant chemotherapy 1 week ago. Pure tone audiometry shows a sensorineural hearing loss of 45 dB. The expected beneficial effect of the drug that caused this patient's symptoms is most likely due to which of the following actions?
+Options:
+A: Inhibition of thymidine synthesis
+B: Inhibition of proteasome
+C: Hyperstabilization of microtubules
+D: Generation of free radicals
+E: Cross-linking of DNA"
 
 messages = [
     {"role": "system", "content": system_prompt},
@@ -102,4 +112,7 @@ print(tokenizer.decode(response, skip_special_tokens=True))
 If you use these weights or the Med-REFL framework in your research, please cite our paper:
 ```
 
-```
+```
+## Star History
+
+[![Star History Chart](https://api.star-history.com/svg?repos=TianYin123/Med-REFL&type=Date)](https://www.star-history.com/#TianYin123/Med-REFL&Date)