Update app.py

app.py

@@ -276,8 +276,8 @@ def forecast_time_series(df, model_type, horizon, max_steps,y_col):
             #             )
     
         with tab_forecast:
-            if model_choice in forecast_results:
-                df_grid = forecast_results[model_choice]
+            if model_type in forecast_results:
+                df_grid = forecast_results[model_type]
                 st.write(df_grid)
                 # grid_table = AgGrid(
                 #                 df_grid,
@@ -542,13 +542,12 @@ def timegpt_fcst():
                     #             )
             
                 with tab_forecast:
-                    if model_choice in forecast_results:
-                        df_grid = forecast_df
-                        st.write(df_grid)
-                        # grid_table = AgGrid(
-                        #                 df_grid,
-                        #                 theme="alpine",
-                        #             )
+                    df_grid = forecast_df
+                    st.write(df_grid)
+                    # grid_table = AgGrid(
+                    #                 df_grid,
+                    #                 theme="alpine",
+                    #             )
 
 
 
@@ -645,13 +644,12 @@ def timegpt_anom():
                     #             )
             
                 with tab_forecast:
-                    if model_choice in forecast_results:
-                        df_grid = forecast_df
-                        st.write(df_grid)
-                        # grid_table = AgGrid(
-                        #                 df_grid,
-                        #                 theme="alpine",
-                        #             )
+                    df_grid = forecast_df
+                    st.write(df_grid)
+                    # grid_table = AgGrid(
+                    #                 df_grid,
+                    #                 theme="alpine",
+                    #             )