Update app.py

app.py

@@ -29,7 +29,11 @@ def predict(gender, ever_married, work_type, residence_type, smoking_status, age
     input_data[cats] = input_data[cats].apply(lambda x: x.cat.codes)    
 
     prediction = m.predict(input_data)
-    return str(prediction)
+    
+    if prediction == [0]:
+        return "didn't have a stroke"
+    return "have a stroke"
+    
 
 
 gender = gr.inputs.Radio(choices=["Male", "Female", "Other"], label="Gender")