fix

app.py

@@ -24,6 +24,8 @@ columns = [
     "SkinCancer"
 ]
 
+def convert_to_int(s):
+    return int(s[:2])
 
 def predict(
     BMI,
@@ -55,7 +57,7 @@ def predict(
                 MentalHealth,
                 DiffWalking,
                 Sex,
-                AgeEstimate,
+                convert_to_int(AgeEstimate),
                 Race+1,
                 Diabetic,
                 PhysicalActivity,
@@ -68,7 +70,10 @@ def predict(
         ]
     )
     pred = rf.predict(data)[0]
-    return pred
+    if pred == 1:
+        return "Has heart disease"
+    else:
+        return "Does not have any heart disease"
 
 
 inputs = [
@@ -92,7 +97,7 @@ inputs = [
             '65-69',
             '70-74',
             '75-79',
-            '80 or older'], type="index", label="Age Category"),
+            '80 or older'], label="Age Category"),
     gr.Dropdown(["White", "Black", "Asian", "American Indian/Alaskan Native", "Hispanic", "Other"], type="index", label="Race"),
     gr.Dropdown(["No", "Yes", "Yes(during pregnancy)", "No, borderline diabetes"], type="index", label="Diebetic"),
     gr.Checkbox(label="Physical Activity"),