Update README.md

README.md

@@ -1,22 +1,22 @@
 ---
 license: apache-2.0
 tags:
-- axolotl
-- generated_from_trainer
-- text-generation-inference
+  - axolotl
+  - generated_from_trainer
+  - text-generation-inference
 base_model: mistralai/Mistral-7B-Instruct-v0.2
 model_type: mistral
 pipeline_tag: text-generation
 model-index:
-- name: Mistral-7B-Retail-v2
-  results: []
+  - name: Mistral-7B-Retail-v1
+    results: []
 ---
 
-# Mistral-7B-Retail-v2
+# Mistral-7B-Retail-v1
 
 ## Model Description
 
-This model, named "Mistral-7B-Retail-v2," is a specially adjusted version of the [mistralai/Mistral-7B-Instruct-v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2). It is fine-tuned to manage questions and provide answers related to retail services.
+This model, named "Mistral-7B-Retail-v1," is a specially adjusted version of the [mistralai/Mistral-7B-Instruct-v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2). It is fine-tuned to manage questions and provide answers related to retail services.
 
 ## Intended Use
 
@@ -28,8 +28,8 @@ This model, named "Mistral-7B-Retail-v2," is a specially adjusted version of the
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model = AutoModelForCausalLM.from_pretrained("bitext-llm/Mistral-7B-Retail-v2")
-tokenizer = AutoTokenizer.from_pretrained("bitext-llm/Mistral-7B-Retail-v2")
+model = AutoModelForCausalLM.from_pretrained("bitext-llm/Mistral-7B-Retail-v1")
+tokenizer = AutoTokenizer.from_pretrained("bitext-llm/Mistral-7B-Retail-v1")
 
 inputs = tokenizer("<s>[INST] How can I return a purchased item? [/INST]", return_tensors="pt")
 outputs = model.generate(inputs['input_ids'], max_length=50)
@@ -38,7 +38,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 
 ## Model Architecture
 
-The "Mistral-7B-Retail-v2" uses the `MistralForCausalLM` structure with a `LlamaTokenizer`. It maintains the setup of the base model but is enhanced to better respond to retail-related questions.
+The "Mistral-7B-Retail-v1" uses the `MistralForCausalLM` structure with a `LlamaTokenizer`. It maintains the setup of the base model but is enhanced to better respond to retail-related questions.
 
 ## Training Data
 
@@ -78,7 +78,7 @@ This model was developed by the Bitext and trained on infrastructure provided by
 
 ## License
 
-This model, "Mistral-7B-Retail-v2", is licensed under the Apache License 2.0 by Bitext Innovations International, Inc. This open-source license allows for free use, modification, and distribution of the model but requires that proper credit be given to Bitext.
+This model, "Mistral-7B-Retail-v1", is licensed under the Apache License 2.0 by Bitext Innovations International, Inc. This open-source license allows for free use, modification, and distribution of the model but requires that proper credit be given to Bitext.
 
 ### Key Points of the Apache 2.0 License