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

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.pt', map_location=torch.device('cpu'))
35
 
36
  model = LoadModel()
37
 
 
31
 
32
  @st.cache
33
  def LoadModel():
34
+ return torch.load('model_full.pt', map_location=torch.device('cpu'))
35
 
36
  model = LoadModel()
37