Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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__),
|
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()
|