Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,6 +8,15 @@ from model_utils import load_model, load_model_ensemble, ensemble_predict
 from preprocess_utils import load_train_features
 from preprocess_utils import preprocess_pipeline as preprocess
 from inference_utils import add_predictions
+
+import sys
+from pathlib import Path
+
+# Ensure repo root is on sys.path so "src.*" imports work on Hugging Face
+ROOT = Path(__file__).resolve().parents[1]  # /app
+if str(ROOT) not in sys.path:
+    sys.path.insert(0, str(ROOT))
+    
 from src.survival_utils import (
     prepare_cox_df,
     fit_cox,