Update README.md

README.md

@@ -4,9 +4,9 @@ language:
 - en
 - multilingual
 widget:
-- text: 'answer: How can I increase the yield of my potato crop?'
+- text: 'Q: How can I increase the yield of my potato crop?'
   example_title: example 1
-- text: 'answer: how do i check for corn maturity?'
+- text: 'Q: how do i check for corn maturity?'
   example_title: example 2
 tags:
 - agriculture
@@ -30,7 +30,7 @@ pipe = pipeline("text2text-generation", model="mrSoul7766/AgriQBot")
 user_query = "How can I increase the yield of my potato crop?"
 
 # Generate response
-answer = pipe(f"answer: {user_query}", max_length=512)
+answer = pipe(f"Q: {user_query}", max_length=512)
 
 # Print the generated answer
 print(answer[0]['generated_text'])
@@ -47,7 +47,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained("mrSoul7766/AgriQBot")
 max_length = 512
 
 # Generate response with question as input
-input_ids = tokenizer.encode("answer: How can I increase the yield of my potato crop?", return_tensors="pt")
+input_ids = tokenizer.encode("Q: How can I increase the yield of my potato crop?", return_tensors="pt")
 output_ids = model.generate(input_ids, max_length=max_length)
 
 # Decode response