Update app.py

app.py

@@ -1,3 +1,11 @@
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+checkpoint = "Shivam29rathore/t5_10k_base"
+
+#tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
+
+
 def summarize(word):
     import os
     data_path = "/tmp/"
@@ -25,7 +33,7 @@ def summarize(word):
     final_summary = []
     for x in range(0,len(text)-1256,1256):
         text_to_summarize= text[x:x+1256]
-        final_summary.append(t5.predict(text_to_summarize))
+        final_summary.append(model.predict(text_to_summarize))
 
     final_list = list(itertools.chain.from_iterable(final_summary))
     final_list = ''.join(final_list)