Spaces:
Sleeping
Sleeping
Update model.py
Browse files
model.py
CHANGED
|
@@ -24,18 +24,6 @@ MODEL_PATH = "/tmp/global_model.h5"
|
|
| 24 |
DATA_PATH = "/tmp/initial_data.csv"
|
| 25 |
|
| 26 |
|
| 27 |
-
# ===============================
|
| 28 |
-
# 8. GLOBAL LOADING FOR FASTAPI
|
| 29 |
-
# ===============================
|
| 30 |
-
# Load model and encoders for FastAPI app
|
| 31 |
-
if os.path.exists(MODEL_PATH):
|
| 32 |
-
ml_model = tf.keras.models.load_model(MODEL_PATH)
|
| 33 |
-
print("✅ Loaded trained model for prediction.")
|
| 34 |
-
else:
|
| 35 |
-
ml_model = load_or_train_model()
|
| 36 |
-
|
| 37 |
-
scaler = joblib.load("scaler.pkl")
|
| 38 |
-
print("✅ Loaded scaler.")
|
| 39 |
|
| 40 |
|
| 41 |
|
|
@@ -363,6 +351,25 @@ def predict_user_input(user_input):
|
|
| 363 |
return predictions
|
| 364 |
|
| 365 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 366 |
# ===============================
|
| 367 |
# 7. EXAMPLE RUN
|
| 368 |
# ===============================
|
|
|
|
| 24 |
DATA_PATH = "/tmp/initial_data.csv"
|
| 25 |
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
|
| 28 |
|
| 29 |
|
|
|
|
| 351 |
return predictions
|
| 352 |
|
| 353 |
|
| 354 |
+
|
| 355 |
+
|
| 356 |
+
# ===============================
|
| 357 |
+
# 8. GLOBAL LOADING FOR FASTAPI
|
| 358 |
+
# ===============================
|
| 359 |
+
# Load model and encoders for FastAPI app
|
| 360 |
+
if os.path.exists(MODEL_PATH):
|
| 361 |
+
ml_model = tf.keras.models.load_model(MODEL_PATH)
|
| 362 |
+
print("✅ Loaded trained model for prediction.")
|
| 363 |
+
else:
|
| 364 |
+
ml_model = load_or_train_model()
|
| 365 |
+
|
| 366 |
+
scaler = joblib.load("scaler.pkl")
|
| 367 |
+
print("✅ Loaded scaler.")
|
| 368 |
+
|
| 369 |
+
|
| 370 |
+
|
| 371 |
+
|
| 372 |
+
|
| 373 |
# ===============================
|
| 374 |
# 7. EXAMPLE RUN
|
| 375 |
# ===============================
|