Update app.py

app.py

@@ -60,7 +60,7 @@ def genreate_intepriable_output(input,dataset="CUB2011", arch="resnet50",seed=12
     TR=get_augmentation(0.1, img_size, False, False, True, True, normalize_params["CUB2011"])
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     if folder is None:
-        folder = Path.home() / f"tmp/{arch}/{dataset}/{seed}/"
+        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")