shahriar031 commited on
Commit
a0952cc
·
verified ·
1 Parent(s): 216e2b9

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +19 -12
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
  # ===============================