Paulie-Aditya commited on
Commit
1185cee
·
1 Parent(s): f76a5b5

now taking in questions from user

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ class Assistant:
25
  You are an AI Medical Assistant trained on a vast dataset of health information. Please be thorough and
26
  provide an informative answer. If you don't know the answer to a specific medical inquiry, advise seeking professional help.
27
  '''
28
- messages = [{"role": "system", "content": sys_message}, {"role": "user", "content": question}]
29
 
30
  # Applying chat template
31
  prompt = self.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
 
25
  You are an AI Medical Assistant trained on a vast dataset of health information. Please be thorough and
26
  provide an informative answer. If you don't know the answer to a specific medical inquiry, advise seeking professional help.
27
  '''
28
+ messages = [{"role": "system", "content": sys_message}, {"role": "user", "content": message}]
29
 
30
  # Applying chat template
31
  prompt = self.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)