atodorov284 commited on
Commit
6d4beac
·
1 Parent(s): 97f0dbd

Update prediction data

Browse files
streamlit_src/models/air_quality_model.py CHANGED
@@ -65,7 +65,12 @@ class AirQualityModel:
65
  """
66
  data = pd.read_csv(PREDICTION_PATH)
67
 
68
- data = data[data["date"] == pd.Timestamp.now().strftime("%Y-%m-%d")]
 
 
 
 
 
69
 
70
  next_three_days = pd.DataFrame(
71
  {
 
65
  """
66
  data = pd.read_csv(PREDICTION_PATH)
67
 
68
+ filtered = data[data["date"] == datetime.today().strftime("%Y-%d-%m")]
69
+
70
+ if len(filtered) == 0:
71
+ filtered = data[data["date"] == datetime.today().strftime("%Y-%m-%d")]
72
+
73
+ data = filtered
74
 
75
  next_three_days = pd.DataFrame(
76
  {