dogoodson commited on
Commit
8f48df3
·
1 Parent(s): dcd121a

update prompt

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ generator = pipeline("text-generation", model="gpt2", max_new_tokens=100, do_sam
85
 
86
  def summarize_notes(patient_id):
87
  treatment_text = get_treatment_plan_from_patient_id(patient_id)
88
- prompt = f"Context: {treatment_text}\nQuestion: What are a few special care instructions for this patient?\nAnswer:"
89
 
90
  result = generator(prompt, return_full_text=False)
91
  return result[0]["generated_text"].strip()
 
85
 
86
  def summarize_notes(patient_id):
87
  treatment_text = get_treatment_plan_from_patient_id(patient_id)
88
+ prompt = f"Based on this treatment plan, summarize key care instructions for the patient in 2-3 sentences: {treatment_text}"
89
 
90
  result = generator(prompt, return_full_text=False)
91
  return result[0]["generated_text"].strip()