Update app.py

app.py

@@ -19,7 +19,7 @@ class Net(nn.Module):
     def forward(self,x):
         return self.layer(x)
 
-@st.cache
+@st.cache(allow_output_mutation=True)
 def GetModelAndTokenizer():
     model = Net()
     model.load_state_dict(torch.load('model.dat', map_location=torch.device('cpu')))