Upload folder using huggingface_hub

app.py

@@ -37,7 +37,15 @@ input_data = pd.DataFrame([{
 
 
 if st.button("Predict Failure"):
-    prediction = model.predict(input_data)[0]
-    result = "Machine Failure" if prediction == 1 else "No Failure"
-    st.subheader("Prediction Result:")
-    st.success(f"The model predicts: **{result}**")
+    try:
+        # Align input columns with model training columns
+        input_data = input_data[model.feature_names_in_]
+
+        prediction = model.predict(input_data)[0]
+        result = "Machine Failure" if prediction == 1 else "No Failure"
+        st.subheader("Prediction Result:")
+        st.success(f"The model predicts: **{result}**")
+
+    except Exception as e:
+        st.error("⚠️ An error occurred during prediction.")
+        st.exception(e)