Update app.py

app.py

@@ -13,7 +13,7 @@ enmo = st.number_input("ENMO", min_value=0.0, max_value=100.0, step=0.01)
 
 if st.button("Detect Sleep State"):
     input_data = np.array([[angle, enmo]])
-    prediction = model.predict(input_data)
+    prediction = model.predict(input_data)[0]
 
     if prediction == 0:
         st.success("Sleep State: **Wakeup**")