Update app.py

app.py

@@ -17,13 +17,14 @@ model_map = {
 }
 
 # Prediction function
-def dis_prediction(model_name, sex, pregnant, TT4, T3, T4U, FTI, TSH):
+def dis_prediction(model_name, sex, pregnant,on_thyroxine, TT4, T3, T4U, FTI, TSH):
     try:
         model = model_map[model_name]
 
         # Convert input to correct types
         sex = int(sex)
         pregnant = int(pregnant)
+        on_thyroxine = int(on_thyroxine)
         TT4 = float(TT4)
         T3 = float(T3)
         T4U = float(T4U)
@@ -44,6 +45,7 @@ demo = gr.Interface(
         gr.Dropdown(["SVM",  "Random Forest"], label="Select Model"),
         gr.Radio([0, 1], label="Sex (0: Female, 1: Male)"),
         gr.Radio([0, 1], label="Pregnant (0: No, 1: Yes)"),
+        gr.Radio([0, 1], label="On Thyroxine (0: No, 1: Yes)"),
         gr.Number(label="TT4"),
         gr.Number(label="T3"),
         gr.Number(label="T4U"),