Update README.md

README.md

@@ -34,7 +34,7 @@ The model is a result of fine-tuning Mistral-7B-v0.1 on a down stream task, in l
 ### Model Description
 
 <!-- Provide a longer summary of what this model is. -->
-dsfsi/mistral-7b-custom_prompt_long_short_31_gpu_days, model, was fine-tuned for 31 GPU days from base model mistralai/Mistral-7B-v0.1. The model was fine-tuned in efforts to improve translation task
+dsfsi/OMT-LR-Mistral7b, model, was fine-tuned for 31 GPU days from base model mistralai/Mistral-7B-v0.1. The model was fine-tuned in efforts to improve translation task
 for large language model in regard to low resourced morphologically rich African languages using custom prompt.
 
 
@@ -64,7 +64,7 @@ be used in the Education industry to teach core subjects in native South African
 ### Direct Use
 
 <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
-You can download the model, dsfsi/mistral-7b-custom_prompt_long_short_31_gpu_days, and prompt it to translate English sentences to Zulu and Xhosa sentences.
+You can download the model, dsfsi/OMT-LR-Mistral7b, and prompt it to translate English sentences to Zulu and Xhosa sentences.
 
 [More Information Needed]
 
@@ -135,7 +135,8 @@ translator("Translate to Zulu: The cow is eating grass.")
 
 #### Training Hyperparameters
 
-- **Training regime:** 
+- **Training regime:**
+```python
 - peft_config = LoraConfig(
         lora_alpha=16,
         lora_dropout=0.05,
@@ -157,7 +158,7 @@ translator("Translate to Zulu: The cow is eating grass.")
         warmup_steps=100,
         lr_scheduler_type="linear",
 )
-
+```
 
 #### Speeds, Sizes, Times [optional]