Update app.py

app.py

@@ -5,7 +5,7 @@ from datasets import load_dataset
 import os
 from src.basic_plot import basic_chart
 from src.map_viz import calling_map_viz
-from src.map_viz_pred import calling_pred_map_viz
+#from src.map_viz_pred import calling_pred_map_viz
 from src.data_ingestion import daily_average
 from src.heatmap import HeatMap
 from src.data_ingestion import remove_previous_view, merge_volumes
@@ -118,9 +118,9 @@ def main():
         starter_variables = [x_train, str(d), pred_hour_choice, pred_view_choice]
         st.plotly_chart(predicted_figure(clf, starter_variables, figs))
 
-        if checkbox_two_pred:
-            st.subheader("Predictive Traffic Map")
-            st.pyplot(calling_pred_map_viz(counts_df))
+        # if checkbox_two_pred:
+        #     st.subheader("Predictive Traffic Map")
+        #     st.pyplot(calling_pred_map_viz(counts_df))
 
 if __name__ == "__main__":
     main()