Spaces:

Sanjayraju30
/

Risk

Sleeping

Sanjayraju30 commited on 9 days ago

Commit

cd000c1

verified ·

1 Parent(s): a2b18be

Update risk_model.py

Files changed (1) hide show

risk_model.py CHANGED Viewed

@@ -7,7 +7,7 @@ MODEL_PATH = "heating_model.pkl"
 DATA_PATH = "mantle_training.csv"
 HISTORY = []
-# Function to train model if missing
 def train_and_save_model():
     data = pd.read_csv(DATA_PATH)
     X = data[["temperature", "duration"]]
@@ -23,7 +23,7 @@ def load_model():
         return train_and_save_model()
     return load(MODEL_PATH)
-# Load on module load
 model = load_model()
 def predict_risk(temp, duration):

 DATA_PATH = "mantle_training.csv"
 HISTORY = []
+# Train the model and save
 def train_and_save_model():
     data = pd.read_csv(DATA_PATH)
     X = data[["temperature", "duration"]]
         return train_and_save_model()
     return load(MODEL_PATH)
+# Load once at startup
 model = load_model()
 def predict_risk(temp, duration):