update readme

README.md

@@ -8,16 +8,18 @@ library_name: peft
 # OriGen: Enhancing RTL Code Generation with Code-to-Code Augmentation and Self-Reflection
 
 ### Introduction
-OriGen is a fine-tuned lora model designed for Verilog code generation. It is trained on top of DeepSeek Coder 7B using datasets generated from code-to-code augmentation and self-reflection.
+OriGen is a fine-tuned lora model designed for Verilog code generation. It is trained on top of DeepSeek Coder 7B using datasets generated from code-to-code augmentation and self-reflection. The datasets can be found in the [origen_instruction](https://huggingface.co/datasets/henryen/origen_instruction).
 
-OriGen_Fix is a fine-tuned lora model designed for fixing syntax errors in Verilog code. It is trained based on OriGen. 
+OriGen_Fix is a fine-tuned lora model designed for fixing syntax errors in Verilog code. It is trained based on OriGen using debug datasets. The datasets can be found in the [origen_debug](https://huggingface.co/datasets/henryen/origen_debug).
 
-
-The models have been uploaded to Hugging Face, and the repository contains the inference scripts. The dataset and data generation flow will be released soon.
+The models have been uploaded to Hugging Face, and the repository contains the inference scripts. The data generation flow will be released soon.
 
 - **Huggingface**: 
     - https://huggingface.co/henryen/OriGen
     - https://huggingface.co/henryen/OriGen_Fix
+- **Dataset**: 
+    - https://huggingface.co/datasets/henryen/origen_instruction
+    - https://huggingface.co/datasets/henryen/origen_debug
 - **Repository**: https://github.com/pku-liang/OriGen
 
 ### Evaluation Results