Spaces:
Sleeping
Sleeping
update prompt
Browse files
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"
|
| 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()
|