Fola-lad commited on
Commit
cd96633
Β·
1 Parent(s): b77223e

anchor model and data paths to repo root via __file__

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +12 -2
src/streamlit_app.py CHANGED
@@ -1,7 +1,14 @@
 
1
  import streamlit as st
2
  import numpy as np
3
  import pandas as pd
4
 
 
 
 
 
 
 
5
  # ── Constants ──────────────────────────────────────────────────────────────
6
 
7
  LABEL_MAP = {
@@ -26,11 +33,14 @@ EXPLANATIONS = {
26
 
27
  @st.cache_resource
28
  def load_model():
 
 
 
29
  try:
30
  import tensorflow as tf
31
  from model_def import FeedForwardNetwork
32
  model = tf.keras.models.load_model(
33
- "model.keras",
34
  custom_objects={"FeedForwardNetwork": FeedForwardNetwork},
35
  )
36
  return model, "ready"
@@ -92,7 +102,7 @@ with tab1:
92
  )
93
 
94
  try:
95
- samples_df = pd.read_csv("data/samples.csv")
96
  feature_cols = [
97
  c for c in samples_df.columns
98
  if c not in ["Activity", "subject"]
 
1
+ import os
2
  import streamlit as st
3
  import numpy as np
4
  import pandas as pd
5
 
6
+ # Paths anchored to the repo root regardless of working directory
7
+ _SRC_DIR = os.path.dirname(os.path.abspath(__file__)) # /app/src
8
+ _REPO_ROOT = os.path.dirname(_SRC_DIR) # /app
9
+ _MODEL_PATH = os.path.join(_REPO_ROOT, "model.keras")
10
+ _SAMPLES_PATH = os.path.join(_REPO_ROOT, "data", "samples.csv")
11
+
12
  # ── Constants ──────────────────────────────────────────────────────────────
13
 
14
  LABEL_MAP = {
 
33
 
34
  @st.cache_resource
35
  def load_model():
36
+ if not os.path.exists(_MODEL_PATH):
37
+ root_files = os.listdir(_REPO_ROOT)
38
+ return None, f"error: model.keras not found at {_MODEL_PATH}. Files in {_REPO_ROOT}: {root_files}"
39
  try:
40
  import tensorflow as tf
41
  from model_def import FeedForwardNetwork
42
  model = tf.keras.models.load_model(
43
+ _MODEL_PATH,
44
  custom_objects={"FeedForwardNetwork": FeedForwardNetwork},
45
  )
46
  return model, "ready"
 
102
  )
103
 
104
  try:
105
+ samples_df = pd.read_csv(_SAMPLES_PATH)
106
  feature_cols = [
107
  c for c in samples_df.columns
108
  if c not in ["Activity", "subject"]