Sahil0605 commited on
Commit
07ba332
·
1 Parent(s): e9cc1b4

first push

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -1
src/streamlit_app.py CHANGED
@@ -14,12 +14,13 @@ st.sidebar.title("Navigation")
14
  page = st.sidebar.selectbox("Go to", ["Upload File", "Preprocess Data", "Train & Select Model", "Create App"])
15
 
16
  # Writable directories for Hugging Face / Streamlit Cloud
17
- BASE_DIR = "/tmp" # only /tmp is writable in HF Spaces
18
  UPLOAD_DIR = os.path.join(BASE_DIR, "uploads")
19
  PROCESSED_DIR = os.path.join(BASE_DIR, "processed")
20
  PKL_DIR = os.path.join(BASE_DIR, "pkl")
21
  ENCODER_DIR = os.path.join(BASE_DIR, "encoders")
22
 
 
23
  for d in [UPLOAD_DIR, PROCESSED_DIR, PKL_DIR, ENCODER_DIR]:
24
  os.makedirs(d, exist_ok=True)
25
 
 
14
  page = st.sidebar.selectbox("Go to", ["Upload File", "Preprocess Data", "Train & Select Model", "Create App"])
15
 
16
  # Writable directories for Hugging Face / Streamlit Cloud
17
+ BASE_DIR = "/tmp"
18
  UPLOAD_DIR = os.path.join(BASE_DIR, "uploads")
19
  PROCESSED_DIR = os.path.join(BASE_DIR, "processed")
20
  PKL_DIR = os.path.join(BASE_DIR, "pkl")
21
  ENCODER_DIR = os.path.join(BASE_DIR, "encoders")
22
 
23
+ # Create all necessary writable directories
24
  for d in [UPLOAD_DIR, PROCESSED_DIR, PKL_DIR, ENCODER_DIR]:
25
  os.makedirs(d, exist_ok=True)
26