Link model to paper and update GitHub repository URL

README.md

@@ -1,21 +1,23 @@
 ---
-license: llama3.1
 base_model: dlutIR/RexDrug-base
 library_name: peft
+license: llama3.1
 pipeline_tag: text-generation
 tags:
-  - drug-combination
-  - relation-extraction
-  - biomedical
-  - llama
-  - chain-of-thought
-  - lora
-  - grpo
+- drug-combination
+- relation-extraction
+- biomedical
+- llama
+- chain-of-thought
+- lora
+- grpo
 ---
 
 # RexDrug-adapter
 
-This is the LoRA adapter for **RexDrug**, trained via GRPO (Group Relative Policy Optimization) on top of [RexDrug-base](https://huggingface.co/dlutIR/RexDrug-base) for biomedical drug combination relation extraction with chain-of-thought reasoning.
+This is the LoRA adapter for **RexDrug**, presented in the paper [RexDrug: Reliable Multi-Drug Combination Extraction through Reasoning-Enhanced LLMs](https://huggingface.co/papers/2603.08166).
+
+It was trained via GRPO (Group Relative Policy Optimization) on top of [RexDrug-base](https://huggingface.co/dlutIR/RexDrug-base) for biomedical drug combination relation extraction with chain-of-thought reasoning.
 
 ## Model Details
 
@@ -45,7 +47,8 @@ model.eval()
 # 2. Prepare input
 messages = [
     {"role": "system", "content": "You are an expert in biomedical drug-drug relation extraction. ..."},
-    {"role": "user",   "content": "Target sentence: ... \nContext paragraph: ..."},
+    {"role": "user",   "content": "Target sentence: ... 
+Context paragraph: ..."},
 ]
 input_text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
 inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
@@ -57,8 +60,8 @@ response = tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:], skip_spe
 print(response)
 ```
 
-See the full example in the [GitHub repository](https://github.com/your-org/RexDrug).
+See the full example and code in the [GitHub repository](https://github.com/DUTIR-BioNLP/RexDrug).
 
 ## License
 
-This model is built upon Llama 3.1 and is subject to the [Llama 3.1 Community License Agreement](https://github.com/meta-llama/llama-models/blob/main/models/llama3_1/LICENSE).
+This model is built upon Llama 3.1 and is subject to the [Llama 3.1 Community License Agreement](https://github.com/meta-llama/llama-models/blob/main/models/llama3_1/LICENSE).