test

app.py

@@ -99,17 +99,14 @@ def make_prediction(alcohol, arrest_type, belts, contributed_to_accident, disobe
 
 iface = gr.Interface(fn=make_prediction,
                      inputs=[gr.components.Checkbox(label='Was the driver under the influence of alcohol?'),
-                             #gr.components.Dropdown(label='Choose the arrest type', choices=tf['OneHotEncoder'].transformers_[0][1].categories_[0], value='A - Marked Patrol'),
-                            'Marked Patrol',
+                             gr.components.Dropdown(label='Choose the arrest type', choices=list(tf['OneHotEncoder'].transformers_[0][1].categories_[0]), value='A - Marked Patrol'), #'Marked Patrol',
                              gr.components.Checkbox(label='Were seatbelts used appropriately?'),
                              gr.components.Checkbox(label='Did the driver actions contribute to an accident?'),
                              gr.components.Checkbox(label='Was the driver disobedient? (such as failing to display documentation upon request)?'),
-                             #gr.components.Dropdown(label='Choose the driver city', choices=tf['OrdinalEncoder_VeryHighCardinality'].categories_[0], value='SILVER SPRING'),
-                             'SILVER SPRING',
+                             'SILVER SPRING', #gr.components.Dropdown(label='Choose the driver city', choices=tf['OrdinalEncoder_VeryHighCardinality'].categories_[0], value='SILVER SPRING'),
                              gr.components.Dropdown(label='Driver Gender', choices=['M', 'F', 'N'], value='M'),
                              gr.components.Checkbox(label='Was the driver driving with Invalid Documentation (such as suspended registration, suspended license, expired registration plates and validation tabs or expired license plate)?'),
-                             #gr.components.Dropdown(label='Vehicle Make', choices=tf['OrdinalEncoder_HighCardinality'].categories_[0], value='TOYOTA'),
-                             'TOYOTA',
+                             'TOYOTA', #gr.components.Dropdown(label='Vehicle Make', choices=tf['OrdinalEncoder_HighCardinality'].categories_[0], value='TOYOTA'),
                              gr.components.Checkbox(label='Was the driver using a mobile phone while driving?'),
                              gr.components.Checkbox(label='Was the driver caught driving with negligence (example switching lanes in an unsafe manner)?'),
                              gr.components.Slider(minimum=1, step=1, label='Number of offences committed'),