Pushpak21 commited on
Commit
fbf66a8
·
verified ·
1 Parent(s): 06105ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -80,7 +80,10 @@ with tab2:
80
  st.dataframe(df.head())
81
 
82
  df = get_predictions(df)
83
- df = reorder_columns(df, ["Product_Store_Sales_Total", "Predicted_Sales"])
 
 
 
84
 
85
  col1, col2 = st.columns([6, 1])
86
  with col1:
 
80
  st.dataframe(df.head())
81
 
82
  df = get_predictions(df)
83
+ if "Product_Store_Sales_Total" in df.columns:
84
+ df = reorder_columns(df, ["Product_Store_Sales_Total", "Predicted_Sales"])
85
+ else:
86
+ df = reorder_columns(df, ["Predicted_Sales"])
87
 
88
  col1, col2 = st.columns([6, 1])
89
  with col1: