Update main.py

main.py

@@ -76,10 +76,11 @@ def get_hail_data(lat, lon, start_date, end_date, radius_miles, get_max):
         data_max = np.max(data_mat, axis=(1, 2))
         df_data = pd.DataFrame({'Date': dates_all,
                                'Hail_max': data_max})
+    # Get all Data
     else:
         data_all = list(data_mat)
         df_data = pd.DataFrame({'Date': dates_all,
-                               'Hail_all': data_mat})
+                               'Hail_all': data_all})
 
     df_data['Date'] = pd.to_datetime(df_data['Date'], format='%Y%m%d')
     df_data = df_data.set_index('Date')
@@ -101,3 +102,4 @@ async def predict(lat: float, lon: float, start_date: str, end_date: str, radius
         results = pd.DataFrame({'Date': ['error'], 'Hail_max': ['error']})
 
     return results.to_json()
+