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 {"HeartDisease": 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"),
@@ -103,13 +108,10 @@ inputs = [
     gr.Checkbox(label="Skin Cancer"),
 ]
 
-
-output = gr.Label(num_top_classes=1)
-
 iface = gr.Interface(
     fn=predict,
     inputs=inputs,
-    outputs=output,
+    outputs="text",
     description="The purpose of this model is to predict whether or not a person has any heart diseases or not.",
 )