Update app.py

app.py

@@ -18,13 +18,13 @@ def predict_and_generate(name, age, gender, platelets, hb, duration, calcium, po
 
     # 1️⃣ Prepare input DataFrame
     data = pd.DataFrame([{
-        "Platelet Count": platelets,
-        "Age": age,
-        "Hemoglobin (g/dL)": hb,
-        "Treatment Duration": duration,
-        "Calcium (mg/dL)": calcium,
-        "Potassium": potassium,
-        "Gender": gender
+    "Platelet Count": platelets,
+    "Age": age,
+    "Hemoglobin (g/dL)": hb,
+    "Treatment Duration": duration,
+    "Calcium (mg/dL)": calcium,
+    "Potassium (mmol/L)": potassium,
+    "Gender": gender
     }])
 
     # 2️⃣ Pass data through preprocessor