jason-moore commited on
Commit
17aca3c
·
1 Parent(s): 69c7720

Trim output

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -65,6 +65,7 @@ PLAN: This section outlines the next steps in the patient's care, including trea
65
  # Decode and extract the response part
66
  response = tokenizer.batch_decode(outputs)[0]
67
  soap_note = response.split("### Response:")[1].strip() if "### Response:" in response else response
 
68
 
69
  return soap_note
70
 
 
65
  # Decode and extract the response part
66
  response = tokenizer.batch_decode(outputs)[0]
67
  soap_note = response.split("### Response:")[1].strip() if "### Response:" in response else response
68
+ soap_note = soap_note.replace("<|end▁of▁sentence|>", "").strip()
69
 
70
  return soap_note
71