Update README.md

README.md

@@ -11,9 +11,9 @@ tags:
 - fine-tuning
 ---
 
-# Model Card for MamaBot-Llama-1
+# Model Card for MamaBot-Llama
 
-MamaBot-Llama-1 is an opensource fine-tuned large language model developed by HelpMum to assist with maternal healthcare by providing accurate and reliable answers to questions about pregnancy and childbirth. The model has been fine-tuned on Llama 3.1 8b-instruct using a dataset of maternal healthcare questions and answers.
+MamaBot-Llama is an opensource fine-tuned large language model developed by HelpMum to assist with maternal healthcare by providing accurate and reliable answers to questions about pregnancy and childbirth. The model has been fine-tuned on Llama 3.1 8b-instruct using a dataset of maternal healthcare questions and answers.
 
 ## Model Details
 
@@ -26,7 +26,7 @@ MamaBot-Llama-1 is an opensource fine-tuned large language model developed by He
 
 ### Model Sources
 
-- **Repository:** [MamaBot-Llama-1 on Hugging Face](https://huggingface.co/HelpMum-Personal/mamabot-llama-1)
+- **Repository:** [MamaBot-Llama-1 on Hugging Face](https://huggingface.co/HelpMumHQ/MamaBot-Llama)
 
 ## Uses
 
@@ -63,8 +63,8 @@ Use the code below to get started with the model.
 !pip install -q -U bitsandbytes
 
 from transformers import AutoModelForCausalLM, AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained('HelpMum-Personal/mamabot-llama-1')
-model = AutoModelForCausalLM.from_pretrained('HelpMum-Personal/mamabot-llama-1')
+tokenizer = AutoTokenizer.from_pretrained('HelpMumHQ/MamaBot-Llama')
+model = AutoModelForCausalLM.from_pretrained('HelpMumHQ/MamaBot-Llama')
 
 def generate_response(user_message):
     tokenizer.chat_template = "{%- for message in messages %}{{ bos_token + '[INST] ' + message['content'] + ' [/INST]' if message['role'] == 'user' else ' ' + message['content'] + ' ' + eos_token }}{%- endfor %}"
@@ -154,13 +154,13 @@ The training and inference were conducted using the Hugging Face Transformers li
   author = {HelpMum},
   title = {MamaBot-Llama-1},
   year = {2024},
-  howpublished = {\url{https://huggingface.co/HelpMum-Personal/mamabot-llama-1}},
+  howpublished = {\url{https://huggingface.co/HelpMumHQ/MamaBot-Llama}},
 }
 ```
 
 **APA:**
 
-HelpMum. (2024). MamaBot-Llama-1. Retrieved from https://huggingface.co/HelpMum-Personal/mamabot-llama-1
+HelpMum. (2024). MamaBot-Llama-1. Retrieved from https://huggingface.co/HelpMumHQ/MamaBot-Llama
 
 ## Model Card Contact