Update app.py

app.py

@@ -264,10 +264,11 @@ def transfer_learning_forecasting():
         if st.sidebar.button("Submit"):
             for model_name, forecast_df in forecast_results.items():
                 plot_forecasts(forecast_df, df, f'{model_name} Forecast for {y_col}')
-    
-        end_time = time.time()  # End timing
-        time_taken = end_time - start_time
-        st.success(f"Time taken for {model_choice} forecast: {time_taken:.2f} seconds")
+
+            end_time = time.time()  # End timing
+            time_taken = end_time - start_time
+            st.success(f"Time taken for {model_choice} forecast: {time_taken:.2f} seconds")
+
 
 def dynamic_forecasting():
     st.title("Dynamic Forecasting")