Sahil0605 commited on
Commit
b9fb731
·
1 Parent(s): 2b24720

first push

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +11 -4
src/streamlit_app.py CHANGED
@@ -12,10 +12,17 @@ from sklearn.metrics import accuracy_score, r2_score
12
  st.sidebar.title("Navigation")
13
  page = st.sidebar.selectbox("Go to", ["Upload File", "Preprocess Data", "Train & Select Model", "Create App"])
14
 
15
- os.makedirs("uploads", exist_ok=True)
16
- os.makedirs("processed", exist_ok=True)
17
- os.makedirs("pkl", exist_ok=True)
18
- os.makedirs("encoders", exist_ok=True) # Folder to save encoders
 
 
 
 
 
 
 
19
 
20
  if page == "Upload File":
21
  st.title("Upload CSV or SQL File")
 
12
  st.sidebar.title("Navigation")
13
  page = st.sidebar.selectbox("Go to", ["Upload File", "Preprocess Data", "Train & Select Model", "Create App"])
14
 
15
+ BASE_DIR = "/tmp" # temp folder for deployment
16
+ UPLOAD_DIR = os.path.join(BASE_DIR, "uploads")
17
+ PROCESSED_DIR = os.path.join(BASE_DIR, "processed")
18
+ PKL_DIR = os.path.join(BASE_DIR, "pkl")
19
+ ENCODER_DIR = os.path.join(BASE_DIR, "encoders")
20
+
21
+ os.makedirs(UPLOAD_DIR, exist_ok=True)
22
+ os.makedirs(PROCESSED_DIR, exist_ok=True)
23
+ os.makedirs(PKL_DIR, exist_ok=True)
24
+ os.makedirs(ENCODER_DIR, exist_ok=True)
25
+
26
 
27
  if page == "Upload File":
28
  st.title("Upload CSV or SQL File")