Update src/map_viz_pred.py

src/map_viz_pred.py

@@ -38,7 +38,7 @@ def calling_pred_map_viz(counts_df1):
         except:
             continue
 
-
+    counts_df1['total'] = counts_df1['car'] + counts_df1['motorcycle'] + counts_df1['large_vehicle']
     counts_df1.loc[:,'date_time'] = pd.to_datetime(counts_df1.loc[:,'date'] + " "+ counts_df1.loc[:,'time'], format='%Y-%m-%d %H:%M:%S')
     counts_df1.loc[:,'hour'] = counts_df1.loc[:,'date_time'].apply(hour_rounder)
     counts_df1.loc[:,'day_name'] = counts_df1.loc[:,'date_time'].dt.day_name()