Update app.py
Browse files
app.py
CHANGED
|
@@ -92,23 +92,6 @@ class AnomalyInput(BaseModel):
|
|
| 92 |
sleep_stage: Annotated[Literal['light_sleep','deep_sleep','rem_sleep'], Field(...)]
|
| 93 |
date: Annotated[datetime, Field(...)]
|
| 94 |
|
| 95 |
-
# ===============================
|
| 96 |
-
# Startup event to download & load models
|
| 97 |
-
# ===============================
|
| 98 |
-
@app.on_event("startup")
|
| 99 |
-
def startup_event():
|
| 100 |
-
global heart_model, heart_features, anomaly_model, anomaly_features
|
| 101 |
-
|
| 102 |
-
HEART_MODEL_PATH = download_from_hf(HEART_MODEL_FILENAME)
|
| 103 |
-
ANOMALY_MODEL_PATH = download_from_hf(ANOMALY_MODEL_FILENAME)
|
| 104 |
-
|
| 105 |
-
heart_model_artifacts = joblib.load(HEART_MODEL_PATH)
|
| 106 |
-
heart_model = heart_model_artifacts['model']
|
| 107 |
-
heart_features = heart_model_artifacts['feature_columns']
|
| 108 |
-
|
| 109 |
-
anomaly_model_artifacts = joblib.load(ANOMALY_MODEL_PATH)
|
| 110 |
-
anomaly_model = anomaly_model_artifacts['model']
|
| 111 |
-
anomaly_features = anomaly_model_artifacts['feature_columns']
|
| 112 |
|
| 113 |
# ===============================
|
| 114 |
# Utility: preprocess features
|
|
|
|
| 92 |
sleep_stage: Annotated[Literal['light_sleep','deep_sleep','rem_sleep'], Field(...)]
|
| 93 |
date: Annotated[datetime, Field(...)]
|
| 94 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 95 |
|
| 96 |
# ===============================
|
| 97 |
# Utility: preprocess features
|