Update app.py

app.py

@@ -2,14 +2,14 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 
 
-checkpoint = "https://huggingface.co/Shivam29rathore/t5_10k_base"
+checkpoint = "myt5.sav"
 
 #load model from drive
-#with open('Shivam29rathore/t5_10k_base', "rb") as f:
-  #t5 =  pickle.load(f)
+with open(checkpoint, "rb") as f:
+    t5 =  pickle.load(f)
 
 #tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
+#model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
 
 
 def summarize(word):