Manu101 commited on
Commit
1c6167f
·
verified ·
1 Parent(s): 3ab85fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ device = torch.device(device)
33
 
34
  try:
35
  model = model_file.get_model().to(device)
36
- checkpoint = torch.load(os.path.join(os.path.dirname(__file__),weights_only=False, "model_00350.pt"), map_location=device)
37
  state_dict = {key.replace("_orig_mod.", ""): value for key, value in checkpoint['model'].items()}
38
  model.load_state_dict(state_dict=state_dict)
39
  model.eval()
 
33
 
34
  try:
35
  model = model_file.get_model().to(device)
36
+ checkpoint = torch.load(os.path.join(os.path.dirname(__file__), "model_00350.pt", weights_only=False), map_location=device)
37
  state_dict = {key.replace("_orig_mod.", ""): value for key, value in checkpoint['model'].items()}
38
  model.load_state_dict(state_dict=state_dict)
39
  model.eval()