Update app.py

app.py

@@ -32,7 +32,7 @@ class devops_model(nn.Module):
 
 @st.cache
 def LoadModel():
-    return torch.load('model.pt')
+    return torch.load('model.pt', map_location=torch.device('cpu'))
 
 model = LoadModel()