Update app.py

app.py

@@ -1,22 +1,15 @@
 import gradio as gr
+import joblib
 import numpy as np
 
-# # 🟢 Option 1: Load model with joblib
-# import joblib
-# model = joblib.load("log_reg_model.pkl")
+# ✅ Load the model correctly using joblib
+model = joblib.load("log_reg_model.pkl")
 
-# 🔵 Option 2: (Use only if model saved with pickle)
-
-import pickle
-with open("log_reg_model.pkl", "rb") as f:
-    model = pickle.load(f)
-
-# Prediction function
+# Define prediction function
 def predict_sleep(step: float, hour: float):
     input_data = np.array([[step, hour]])
     prediction = model.predict(input_data)[0]
-    label = "Sleep Onset" if prediction == 1 else "Wakeup"
-    return label
+    return "Sleep Onset" if prediction == 1 else "Wakeup"
 
 # Gradio interface
 iface = gr.Interface(