Update app.py

app.py

@@ -66,7 +66,7 @@ def main():
             prediction = generate_prediction(np.array([[age, income, family_size, pin, Gender, Review, Marital_status, occupation, education]]))
         
             # Show the prediction
-            st.write("Prediction:", prediction[1][1])
+            st.write("Prediction:", prediction[0][1])
 
     elif choice == "About":
         st.subheader("About this app")