Update app.py

app.py

@@ -62,8 +62,8 @@ def genreate_intepriable_output(input,dataset="CUB2011", arch="resnet50",seed=12
     if folder is None:
         folder = Path(f"tmp/{arch}/{dataset}/{seed}/")
     
-    state_dict = torch.load(folder / f"{model_type}_{n_features}_{n_per_class}_FinetunedModel.pth")
-    selection= torch.load(folder / f"SlDD_Selection_50.pt")
+    state_dict = torch.load(folder / f"{model_type}_{n_features}_{n_per_class}_FinetunedModel.pth", map_location=torch.device('cpu'))
+    selection= torch.load(folder / f"SlDD_Selection_50.pt", map_location=torch.device('cpu'))
     state_dict['linear.selection']=selection
     
     feature_sel, sparse_layer, current_mean, current_std, bias_sparse = extract_sel_mean_std_bias_assignemnt(state_dict)