jonathanjordan21 commited on
Commit
e23b346
·
verified ·
1 Parent(s): 5bd0556

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ class NegBinomialModel(nn.Module):
25
 
26
 
27
  model = NegBinomialModel(12)
28
- model.load_state_dict(torch.load("model_weights.pt", map_location='cpu'))
29
  model.eval()
30
 
31
  MU_BANKS = 2.6035915713614286
 
25
 
26
 
27
  model = NegBinomialModel(12)
28
+ model.load_state_dict(torch.load("model_weights(1).pt", map_location='cpu'))
29
  model.eval()
30
 
31
  MU_BANKS = 2.6035915713614286