Update app.py

app.py

@@ -17,10 +17,10 @@ tokenizer = AutoTokenizer.from_pretrained('t5-small')
 #model = torch.load(model_name+'/model.pt')
 
 #if model_name == "T5 Small":
-#model = AutoModelForSeq2SeqLM.from_pretrained(model_name, checkpoint_file="t5_epoch9.ckpt")
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name, checkpoint_file="t5_epoch9.ckpt")
     #model = torch.load(model_name+'/model.pt')
 checkpoint_path=model_name+'/t5_epoch9.ckpt'
-model=T5.load_from_checkpoint(checkpoint_path, map_location=torch.device('cpu'))
+#model=T5.load_from_checkpoint(checkpoint_path, map_location=torch.device('cpu'))
     
 #else:
     #model = GPT2().from_pretrained(model_name)