Update README.md

README.md

@@ -49,10 +49,10 @@ model_pretrained.config.pad_token_id = tokenizer.eos_token_id
 pipe = pipeline(task="text-generation", model=model_pretrained, tokenizer=tokenizer, max_length=100)
 
 def build_prompt(question):
-    prompt = f"[INST]@Enlighten. [/INST] {question}"
-    return prompt
+  prompt=f"[INST]@Enlighten. {question} [/INST]"
+  return prompt
 
-question = "Are my symptoms due to HIV infection? I had a high-risk exposure 15 months ago"
+question = "What does abutment of the nerve root mean?"
 prompt = build_prompt(question)
 
 # Generate text based on the prompt
@@ -66,6 +66,39 @@ print(generated_text)
 
 
 ```
+you will get somethinng like
+```
+Please help. For more information consult an internal medicine physician online ➜ http://iclinic.com/e/gastroenterologist-online-consultation.php.
+```
+
+
+
+also you can 
+
+```python
+def ask(question):
+  promptEnding = "[/INST]"
+  # Guide for answering questions
+  testGuide = 'Answer the following question, at the end of your response say thank you for your query.\n'
+  # Build the question prompt
+  question = testGuide + question + "\n"
+  print(question)
+  # Build the prompt
+  prompt = build_prompt(question)
+  # Generate answer
+  result = pipe(prompt)
+  llmAnswer = result[0]['generated_text']
+  # Remove the prompt from the generated answer
+  index = llmAnswer.find(promptEnding)
+  llmAnswer = llmAnswer[len(promptEnding) + index:]
+  print("LLM Answer:")
+  print(llmAnswer)
+
+question = "For how long should I take Kalachikai powder to overcome PCOD problem?"
+ask(question)
+```
+
+
 
 ## Training Data
 - **Dataset Name:** AI Medical Chatbot