Update src/streamlit_app.py

src/streamlit_app.py

@@ -80,7 +80,7 @@ class MyModel(nn.Module):
         return x
 # Load model
 model = MyModel(num_classes=num_classes).to(device)
-model.load_state_dict(torch.load("brainCNNpytorch_model", map_location=torch.device('cpu')))
+model.load_state_dict(torch.load("src/brainCNNpytorch_model", map_location=torch.device('cpu')))
 model.eval()
 # Label dictionary
 label_dict = {0: "Meningioma", 1: "Glioma", 2: "No Tumor", 3: "Pituitary"}