Update README.md

README.md

@@ -30,7 +30,7 @@ Falcon-1B-Mental-Health-Advisor is a fine-tuned version of the tiiuae/falcon-rw-
 # Important Note
 Mental Health is a sensitive topic. Preferably, use the code snippet provided below in order to get optimal results.
 
-# Falcon-1B Fine-Tuned for Mental Health (LoRA)
+# Falcon-RW-1B Fine-Tuned for Mental Health (LoRA)
 
 This is a LoRA adapter for the Falcon-RW-1B model. It was fine-tuned on the 'marmikpandya/mental-health' dataset.
 
@@ -52,7 +52,7 @@ import re
 import language_tool_python
 
 base_model = "tiiuae/falcon-rw-1b"
-peft_model = "ShivomH/Falcon-1B-Finetuned-Mental-Health"
+peft_model = "ShivomH/Falcon-1B-Mental-Health-v1"
 
 # Load the base model (without LoRA weights initially)
 model = AutoModelForCausalLM.from_pretrained(