Update app.py

app.py

@@ -1,11 +1,17 @@
 import gradio as gr
-import joblib
 import numpy as np
 
-# Load the model using joblib
-model = joblib.load("log_reg_model.pkl")
+# # 🟢 Option 1: Load model with joblib
+# import joblib
+# model = joblib.load("log_reg_model.pkl")
 
-# Define prediction function
+# 🔵 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
 def predict_sleep(step: float, hour: float):
     input_data = np.array([[step, hour]])
     prediction = model.predict(input_data)[0]
@@ -24,6 +30,5 @@ iface = gr.Interface(
     description="Enter step count and hour to predict whether it's sleep onset or wakeup."
 )
 
-# Launch app
 if __name__ == "__main__":
     iface.launch()