Update app.py

app.py

@@ -8,7 +8,7 @@ def summarize_function(notes):
     input_ids = notes_input.input_ids
     gen_tokens = model.generate(input_ids, do_sample=True, temperature = 0.5, max_length=max_length)
     gen_text = tokenizer.batch_decode(gen_tokens)[0]
-    return gen_text.replace(notes, '')
+    return gen_text[len(notes):]
 
 st.markdown("<h1 style='text-align: center; color: #489DDB;'>GPT Clinical Notes Summarizer</h1>", unsafe_allow_html=True)
 st.markdown("<h6 style='text-align: center; color: #489DDB;'>by Bryan Mildort</h1>", unsafe_allow_html=True)