Update app.py

app.py

@@ -1,8 +1,14 @@
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-checkpoint = "Shivam29rathore/t5_10k_base"
 
-#tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+
+checkpoint = "Shivam29rathore/t5_10k_base.sav"
+
+#load model from drive
+#with open('Shivam29rathore/t5_10k_base', "rb") as f:
+  #t5 =  pickle.load(f)
+
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
 model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)