Update app.py

app.py

@@ -11,10 +11,10 @@ st.write("Predict sleep state (`onset` or `wakeup`) using step count and hour.")
 
 # Input Features
 step = st.number_input("Step count:", min_value=0, max_value=10000, value=0)
-hour = st.slider("Hour of day (0–23):", min_value=0, max_value=23, value=0)
+hour = st.number_input("Hour of day (0–23):", min_value=0, max_value=23, value=0)
 
 # Predict Button
 if st.button("Predict Sleep State"):
     input_data = np.array([[step, hour]])
     prediction = model.predict(input_data)[0]
-    st.success(f"Predicted Sleep State: **{prediction}**")
+    return "Sleep Onset" if prediction == 1 else "Wakeup"