Zhe-Zhang commited on
Commit
6863ba8
·
verified ·
1 Parent(s): ca6d464

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ model = nn.Sequential(
73
  nn.ReLU(),
74
  nn.Linear(50, nbr_classes)
75
  )
76
- model.load_state_dict(torch.load("nld.pth", map_location="cpu"))
77
  model.eval()
78
 
79
  # --- prediction function ---
 
73
  nn.ReLU(),
74
  nn.Linear(50, nbr_classes)
75
  )
76
+ model.load_state_dict(torch.load("nld(1).pth", map_location="cpu"))
77
  model.eval()
78
 
79
  # --- prediction function ---