Update app.py

app.py

@@ -1,15 +1,9 @@
 import gradio as gr
-import pickle
+import joblib
 import numpy as np
 
-# store the pickle data in a file named 'payload.pkl'
-with open('log_reg_model.pkl', 'wb') as f:
-    pickle.dump(var, f)
-
-# disassemble the pickle
-# and print the instructions to the command line
-with open('log_reg_model.pkl', 'rb') as f:
-    pickletools.dis(f)
+# Load the model using joblib
+model = joblib.load("log_reg_model.pkl")
 
 # Define prediction function
 def predict_sleep(step: float, hour: float):
@@ -23,7 +17,7 @@ iface = gr.Interface(
     fn=predict_sleep,
     inputs=[
         gr.Number(label="Step Count"),
-        gr.Number(label="Hour (0-23)")
+        gr.Number(label="Hour (0–23)")
     ],
     outputs=gr.Text(label="Prediction"),
     title="Sleep Prediction (Onset/Wakeup)",