Update app.py

app.py

@@ -19,10 +19,15 @@ class Net(nn.Module):
     def forward(self,x):
         return self.layer(x)
 
-@st.cache(allow_output_mutation=True)
-def GetModelAndTokenizer():
+@st.cache
+def GetModel():
     model = Net()
     model.load_state_dict(torch.load('model.dat', map_location=torch.device('cpu')))
+    return model
+
+#@st.cache(allow_output_mutation=True)
+def GetModelAndTokenizer():
+    model = GetModel()
     tokenizer = AutoTokenizer.from_pretrained("Callidior/bert2bert-base-arxiv-titlegen")
     model_emb = AutoModelForSeq2SeqLM.from_pretrained("Callidior/bert2bert-base-arxiv-titlegen")
     return model, tokenizer, model_emb