mbabanov commited on
Commit
4b5ef8f
1 Parent(s): e1f65e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ class devops_model(nn.Module):
31
 
32
  @st.cache
33
  def LoadModel():
34
- return torch.load('model_full.pt', map_location=torch.device('cpu'))
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