azrai99 commited on
Commit
ba185a4
·
verified ·
1 Parent(s): b895dd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -236,7 +236,6 @@ def transfer_learning_forecasting():
236
  st.session_state.uploaded_file = uploaded_file
237
  else:
238
  uploaded_file = st.session_state.uploaded_file
239
- df = pd.read_csv(uploaded_file)
240
  st.session_state.df = df
241
 
242
  else:
@@ -251,6 +250,8 @@ def transfer_learning_forecasting():
251
  columns.pop(columns.index('ds'))
252
  columns.pop(columns.index('unique_id'))
253
  opt = columns
 
 
254
  y_col = st.selectbox("Select Target column", options=opt, index=0)
255
 
256
  st.session_state.ds_col = ds_col
 
236
  st.session_state.uploaded_file = uploaded_file
237
  else:
238
  uploaded_file = st.session_state.uploaded_file
 
239
  st.session_state.df = df
240
 
241
  else:
 
250
  columns.pop(columns.index('ds'))
251
  columns.pop(columns.index('unique_id'))
252
  opt = columns
253
+ if 'ds' in opt:
254
+ opt.remove('ds')
255
  y_col = st.selectbox("Select Target column", options=opt, index=0)
256
 
257
  st.session_state.ds_col = ds_col