Update README.md

README.md

@@ -7,6 +7,8 @@ license: apache-2.0
 base_model:
 - mistralai/Mistral-Nemo-Instruct-2407
 pipeline_tag: reinforcement-learning
+datasets:
+- gretelai/synthetic_text_to_sql
 ---
 # Model Card: Mistral-Nemo-Instruct-2407_ORPO- Fine-Tuned for Text-to-SQL
 
@@ -79,9 +81,9 @@ from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 from peft import PeftConfig,PeftModel
 
 # Load the fine-tuned peft model
-peft_config = PeftConfig.from_pretrained(model_path) 
+peft_config = PeftConfig.from_pretrained("JHuel/Mistral-Nemo-Instruct-2407_DPO_qlora") 
 model = AutoModelForCausalLM.from_pretrained(peft_config.base_model_name_or_path)
-model = PeftModel.from_pretrained(model, model_path)
+model = PeftModel.from_pretrained(model, "JHuel/Mistral-Nemo-Instruct-2407_DPO_qlora")
 
 
 # Load the fine-tuned model