Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import torch
 st.title("TextPressoMachine")
 
-#from transformers import AutoModelForSeq2SeqLM
+from transformers import AutoModelForSeq2SeqLM
 from t5_model import T5
 from transformers import AutoTokenizer
 from transformers import pipeline
@@ -17,7 +17,7 @@ tokenizer = AutoTokenizer.from_pretrained('t5-small')
 #model = torch.load(model_name+'/model.pt')
 
 #if model_name == "T5 Small":
-model = T5().from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name, checkpoint_file="t5_epoch9.ckpt"))
     #model = torch.load(model_name+'/model.pt')