Update app.py

app.py

@@ -9,7 +9,7 @@ from transformers import SamModel, SamProcessor
 # Load the processor and model
 processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
 model = SamModel.from_pretrained("facebook/sam-vit-base")
-model_path = "SAM/mito_model_checkpoint.pth"
+model_path = "mito_model_checkpoint.pth"
 model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)