Ciroc0 commited on
Commit
0a1149f
·
1 Parent(s): c8b3dbb

Fix frontend snapshot publishing

Browse files
Files changed (2) hide show
  1. app.py +8 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -33,6 +33,14 @@ APP_NAME = "dmi-collector"
33
 
34
  faulthandler.enable()
35
 
 
 
 
 
 
 
 
 
36
  # Extended feature set from PLAN.md
37
  FORECAST_FEATURES = [
38
  "temperature_2m",
 
33
 
34
  faulthandler.enable()
35
 
36
+ MODEL_FILES = {
37
+ "temperature": "temperature_models.pkl",
38
+ "wind_speed": "wind_speed_models.pkl",
39
+ "wind_gust": "wind_gust_models.pkl",
40
+ "rain_event": "rain_event_models.pkl",
41
+ "rain_amount": "rain_amount_models.pkl",
42
+ }
43
+
44
  # Extended feature set from PLAN.md
45
  FORECAST_FEATURES = [
46
  "temperature_2m",
requirements.txt CHANGED
@@ -6,4 +6,5 @@ requests>=2.31.0
6
  schedule>=1.2.0
7
  pyarrow>=15.0.0
8
  joblib>=1.3.0
 
9
  tzdata>=2024.1
 
6
  schedule>=1.2.0
7
  pyarrow>=15.0.0
8
  joblib>=1.3.0
9
+ xgboost-cpu==3.2.0
10
  tzdata>=2024.1