Update src/streamlit_app.py

src/streamlit_app.py

@@ -330,8 +330,9 @@ def prepare_test_image(img, dim=150):
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-model_path = os.path.join(os.getcwd(), 'Hyper_U_NET_pytorch-MAE-30Epoch.pth')
+# model_path = os.path.join(os.getcwd(), 'Hyper_U_NET_pytorch-MAE-30Epoch.pth')
 # model_path = "Hyper_U_NET_pytorch-MAE-30Epoch.pth"
+model_path = '/app/src/Hyper_U_NET_pytorch-MAE-30Epoch.pth'
 
 test_model = load_model_for_inference(model_path, device)