Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import re
 
 def summarize_function(notes):
-    gen_text = pipe(notes, max_length=(len(notes.split(' '))*2*1.225), temperature=0.8, num_return_sequences=1, top_p=0.2)[0]['generated_text'][len(notes):]
+    gen_text = pipe(notes, max_new_tokens=(len(notes.split(' '))*2*.215), temperature=0.8, num_return_sequences=1, top_p=0.2)[0]['generated_text'][len(notes):]
     for i in range(len(gen_text)):
         if gen_text[-i-8:].startswith('[Notes]:'):
             gen_text = gen_text[:-i-8]