Spaces:

stal76
/

hw2

Runtime error

stal76 commited on Dec 24, 2022

Commit

1480d1d

1 Parent(s): d1e1d59

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ def GetModel():
     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.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")