Update app.py

app.py

@@ -6,7 +6,19 @@ model = joblib.load('modelB.joblib')
 unique_values = joblib.load('unique_valuesB.joblib')
     
 unique_gender =  unique_values["Gender"]
-
+def BMI_Cat(BMI):
+    if BMI == 0:
+        return "Extremly Weak"
+    elif BMI == 1:
+        return "Weak"
+    elif BMI == 2:
+        return "Normal"
+    elif BMI == 3:
+        return "Overweight"
+    elif BMI == 4:
+        return "Obesity"
+    elif BMI == 5:
+        return "Extremly Obesity"
 
 def main():
     st.title("BMI Predict")
@@ -21,6 +33,7 @@ def main():
             result=model.predict(pd.DataFrame({"Gender": [Gender],
                                                "Height": [Height],
                                                "Weight": [Weight]}))
+            result= BMI_Cat(result[0])
             st.success('The predicted BMI is {}'.format(result))
 
 if __name__=='__main__':