Spaces:

Elegbede
/

Lung-Cancer-Prediction

Runtime error

Elegbede commited on May 3, 2024

Commit

51274fc

verified ·

1 Parent(s): 14a162e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ def predict(gender, smoking, yellow_fingers, anxiety, peer_pressure,
                   'SHORTNESS OF BREATH': shortness_of_breath,
                   'SWALLOWING DIFFICULTY': swallowing_difficulty,
                   'CHEST PAIN': chest_pain, 'AGE': age}
-    input_df = pd.DataFrame.from_dict([input_dict])
     dtest = xgb.DMatrix(input_df)

                   'SHORTNESS OF BREATH': shortness_of_breath,
                   'SWALLOWING DIFFICULTY': swallowing_difficulty,
                   'CHEST PAIN': chest_pain, 'AGE': age}
+    input_df = pd.DataFrame.from_dict([input_dict]).astype("int")
     dtest = xgb.DMatrix(input_df)