Add library name

README.md

@@ -1,9 +1,9 @@
 ---
-license: apache-2.0
+base_model:
+- Qwen/Qwen2.5-7B-Instruct
 language:
 - en
-base_model:
-- Qwen/Qwen2.5-7B-Instruct 
+license: apache-2.0
 pipeline_tag: question-answering
 tags:
 - medical
@@ -11,6 +11,7 @@ tags:
 - lora
 - dpo
 - reflection
+library_name: transformers
 ---
 
 <div align="center">
@@ -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},
@@ -101,5 +111,8 @@ 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)