update readme

README.md

@@ -7,10 +7,12 @@ library_name: peft
 
 # OriGen: Enhancing RTL Code Generation with Code-to-Code Augmentation and Self-Reflection
 
+# 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. The datasets can be found in the [origen_instruction](https://huggingface.co/datasets/henryen/origen_instruction).
+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_dataset_instruction](https://huggingface.co/datasets/henryen/origen_dataset_instruction).
 
-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).
+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_dataset_debug](https://huggingface.co/datasets/henryen/origen_dataset_debug).
 
 The models have been uploaded to Hugging Face, and the repository contains the inference scripts. The data generation flow will be released soon.
 
@@ -18,8 +20,8 @@ The models have been uploaded to Hugging Face, and the repository contains the i
     - 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
+    - https://huggingface.co/datasets/henryen/origen_dataset_instruction
+    - https://huggingface.co/datasets/henryen/origen_dataset_debug
 - **Repository**: https://github.com/pku-liang/OriGen
 
 ### Evaluation Results