Upload folder using huggingface_hub

app.py

@@ -71,8 +71,17 @@ input_data = pd.DataFrame([{
 if st.button("Predict Engine Condition"):
     try:
         prediction = ada_model.predict(input_data)[0]
+        
+        # Map numeric prediction to labels
+        if prediction == 0:
+            result = "Good ✅"
+        elif prediction == 1:
+            result = "Faulty ❌"
+        else:
+            result = f"Unknown ({prediction})"
+
         st.subheader("Prediction Result")
-        st.success(f"Estimated Engine Condition: **{prediction}**")
+        st.success(f"Estimated Engine Condition: **{result}**")
+        
     except Exception as e:
         st.error(f"❌ Error during prediction: {str(e)}")
-