Update app.py

app.py

@@ -86,8 +86,7 @@ with st.expander("🔹 **Medical Information**", expanded=True):
 
 if st.button("🔍 Predict Risk"):
     try:
-        input_data = [[Age,BMI,Menstrual_Irregularity,Testosterone_Level,Antral_Follicle_Count]]
-
+        input_df = pd.DataFrame([[Age,BMI,Menstrual_Irregularity, Testosterone_Level,Antral_Follicle_Count]], columns=['Age','BMI','Menstrual_Irregularity','Testosterone_Level','Antral_Follicle_Count'])
         prediction = model.predict(input_data)[0]  
         probability = model.predict_proba(input_data)[0][1] * 100