Spaces:

Haojiacheng
/

CSF_Seg

Runtime error

Haojiacheng commited on Nov 6, 2023

Commit

e396124

1 Parent(s): 0873514

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -103,7 +103,7 @@ def process_nii_file(input_nii_file, slice, mode):
             res_block=True,
             dropout_rate=0.0,
         ).to(device)
-        model.load_state_dict(torch.load(os.path.join(root_dir, "best_metric_model67v2.pth")，map_location=torch.device('cpu'))) # CPU map
         test_transforms = Compose(
             [

             res_block=True,
             dropout_rate=0.0,
         ).to(device)
+        model.load_state_dict(torch.load(os.path.join(root_dir, "best_metric_model67v2.pth"), map_location=torch.device('cpu'))) # CPU map
         test_transforms = Compose(
             [