degide commited on
Commit
cbce739
·
verified ·
1 Parent(s): 4da3949

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -61,7 +61,11 @@ def generate_medical_response(message, history):
61
  "- How is high blood pressure diagnosed?"
62
  )
63
 
64
- prompt = f"<|system|>\nYou are a highly accurate and helpful medical assistant.</s>\n<|user|>\n{message}</s>\n<|assistant|>\n"
 
 
 
 
65
  inputs = tokenizer(prompt, return_tensors="pt")
66
 
67
  with torch.no_grad():
 
61
  "- How is high blood pressure diagnosed?"
62
  )
63
 
64
+ prompt = (
65
+ f"<|system|>\nYou are a highly accurate and helpful medical assistant.</s>\n"
66
+ f"<|user|>\n{message}</s>\n"
67
+ f"<|assistant|>\n"
68
+ )
69
  inputs = tokenizer(prompt, return_tensors="pt")
70
 
71
  with torch.no_grad():