Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import torch
 st.title("TextPressoMachine")
 
 #from transformers import AutoModelForSeq2SeqLM
@@ -15,7 +16,8 @@ model_name = models[selected_model]
 tokenizer = AutoTokenizer.from_pretrained('t5-small')
 
 if model_name == "T5 Small":
-    model = T5().from_pretrained(model_name)
+    #model = T5().from_pretrained(model_name)
+    model = torch.load(model_name+'/model.pt')
 
     
 #else: