Zhe-Zhang commited on
Commit
666e224
·
verified ·
1 Parent(s): 50bcdf5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ model = nn.Sequential(
66
  nn.ReLU(),
67
  nn.Linear(50, num_classes)
68
  )
69
- model.load_state_dict(torch.load("nld.pth", map_location="cpu"))
70
  model.eval()
71
 
72
  # --- prediction ---
 
66
  nn.ReLU(),
67
  nn.Linear(50, num_classes)
68
  )
69
+ model.load_state_dict(torch.load("nld (1).pth", map_location="cpu"))
70
  model.eval()
71
 
72
  # --- prediction ---