Update app.py

app.py

@@ -1,7 +1,6 @@
 import pickle
 import gradio as gr
 
-# Cargar el modelo y el vectorizador
 with open('models/risk-model.pck', 'rb') as f:
     dv, model = pickle.load(f)
 
@@ -38,7 +37,6 @@ def predict_risk(
         "medical_unit": medical_unit,
     }
 
-    # Predecir el riesgo con el modelo
     risk, prediction = predict_single(customer, dv, model)
 
     return {
@@ -54,6 +52,7 @@ with gr.Blocks() as interface:
 
     with gr.Row():
         with gr.Column():
+            gr.Markdown("### Inputs")
             sex = gr.Dropdown(
                 choices=["Femenino", "Masculino"],
                 label="Sexo",
@@ -76,9 +75,10 @@ with gr.Blocks() as interface:
                 maximum=3,
                 info="1-3 means the patient was diagnosed with COVID; 4+ means not diagnosed.",
             )
-            patient_type = gr.Radio(
-                choices=[1, 2],
+            patient_type = gr.Dropdown(
+                choices=["1", "2"],
                 label="Patient Type",
+                value="1",
                 interactive=True,
                 info="1 for Returned Home, 2 for Hospitalization",
             )
@@ -130,4 +130,4 @@ with gr.Blocks() as interface:
     )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()