Spaces:

ngaggion
/

Chest-x-ray-HybridGNet-Segmentation

Running

ngaggion commited on Aug 27, 2023

Commit

e736992

1 Parent(s): c7f8e55

Update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -67,7 +67,7 @@ def loadModel(device):
     config['skip_features'] = f
     hybrid = Hybrid(config.copy(), D_t, U_t, A_t).to(device)
-    hybrid.load_state_dict(torch.load("weights/weights.pt"))
     hybrid.eval()
     return hybrid

     config['skip_features'] = f
     hybrid = Hybrid(config.copy(), D_t, U_t, A_t).to(device)
+    hybrid.load_state_dict(torch.load("weights/weights.pt", map_location=torch.device(device)))
     hybrid.eval()
     return hybrid