Update src/streamlit_app.py

src/streamlit_app.py

@@ -82,7 +82,7 @@ class MyModel(nn.Module):
 # Load model
 model = MyModel(num_classes=num_classes).to(device)
 try:
-    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')))
 except FileNotFoundError:
     st.error("Model file 'brainCNNpytorch_model' not found. Please upload the file correctly.")
     st.stop()