Update README.md

README.md

@@ -22,24 +22,42 @@ This model was trained using AutoTrain. For more information, please visit [Auto
 
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_path = "PATH_TO_THIS_REPO"
+model_path = "kr-manish/Mistral-7B-autotrain-finetune-QA-vx"
+tokenizer = AutoTokenizer.from_pretrained(model_path)
+model = AutoModelForCausalLM.from_pretrained(model_path)
+
+input_text = "What is the scientific name of the honey bee?"
+# Tokenize input text
+input_ids = tokenizer.encode(input_text, return_tensors="pt")
+
+# Generate output text
+output = model.generate(input_ids, max_length=100, num_return_sequences=1, do_sample=True)
+
+# Decode and print output
+generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
+print(generated_text)
+
+#Apis mellifera.
+```
+
+# Usage
+
+```python
+
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
+model_path = "kr-manish/Mistral-7B-autotrain-finetune-QA-vx"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForCausalLM.from_pretrained(
-    model_path,
-    device_map="auto",
-    torch_dtype='auto'
-).eval()
-
-# Prompt content: "hi"
-messages = [
-    {"role": "user", "content": "hi"}
-]
-
-input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
-output_ids = model.generate(input_ids.to('cuda'))
-response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
-
-# Model response: "Hello! How can I assist you today?"
-print(response)
+model = AutoModelForCausalLM.from_pretrained(model_path)
+
+# Generate response
+input_text = "Give three tips for staying healthy."
+input_ids = tokenizer.encode(input_text, return_tensors="pt")
+output = model.generate(input_ids, max_new_tokens = 200)
+predicted_text = tokenizer.decode(output[0], skip_special_tokens=True)
+print(predicted_text)
+
+#Give three tips for staying healthy. [/INST] 1. Eat a balanced diet: Make sure to include plenty of fruits, vegetables, lean proteins, and whole grains in your diet. Avoid processed foods and sugary drinks. 
+#2. Exercise regularly: Aim for at least 30 minutes of moderate exercise every day, such as walking, cycling, or swimming. 
+#3. Get enough sleep: Aim for 7-9 hours of quality sleep each night. Make sure to create a relaxing bedtime routine and stick to a regular sleep schedule. 
 ```