Update app.py

app.py

@@ -2,9 +2,9 @@ import streamlit as st
 
 def summarize_function(notes):
     notes_input = tokenizer(notes, return_tensors='pt')
-    # output = model(**notes_input)
-    # max_length = len(output['logits'][0]) + 35
-    max_length = len(notes.split(' ')) + 35
+    output = model(**notes_input)
+    max_length = len(output['logits'][0]) + 35
+    # max_length = len(notes.split(' ')) + 35
     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]