Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ class devops_model(nn.Module):
|
|
31 |
|
32 |
@st.cache
|
33 |
def LoadModel():
|
34 |
-
return torch.load('
|
35 |
|
36 |
model = LoadModel()
|
37 |
|
|
|
31 |
|
32 |
@st.cache
|
33 |
def LoadModel():
|
34 |
+
return torch.load('model.pt', map_location=torch.device('cpu'))
|
35 |
|
36 |
model = LoadModel()
|
37 |
|