Update app.py

app.py

@@ -48,12 +48,12 @@ def genreate_intepriable_output(input,dataset="CUB2011", arch="resnet50",seed=12
   
     model = get_model(arch, n_classes, reduced_strides)
     tr=transforms.ToTensor()
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    device = torch.device("cpu")
     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)