Update app.py

app.py

@@ -99,26 +99,26 @@ 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=list(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'), 
                              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)?'),
-                             'SILVER SPRING', #gr.components.Dropdown(label='Choose the driver city', choices=tf['OrdinalEncoder_VeryHighCardinality'].categories_[0], value='SILVER SPRING'),
+                             gr.components.Dropdown(label='Choose the driver city', choices=list(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)?'),
-                             'TOYOTA', #gr.components.Dropdown(label='Vehicle Make', choices=tf['OrdinalEncoder_HighCardinality'].categories_[0], value='TOYOTA'),
+                             gr.components.Dropdown(label='Vehicle Make', choices=list(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'),
                              gr.components.Checkbox(label='Did the violation involve any personal injury?'),
                              gr.components.Checkbox(label='Did the violation involve any property damage?'),
-                             gr.components.Dropdown(label='Choose the race of the driver', choices=tf['OneHotEncoder'].transformers_[0][1].categories_[2], value='WHITE'),
+                             gr.components.Dropdown(label='Choose the race of the driver', choices=list(tf['OneHotEncoder'].transformers_[0][1].categories_[2]), value='WHITE'),
                              gr.components.Checkbox(label='Did the driver fail to obey signs and markings (such as traffic control device instructions, stop lights, red signal and stop sign lines)?'),
-                             gr.components.Dropdown(label='Choose the race of the driver', choices=tf['OneHotEncoder'].transformers_[0][1].categories_[3], value='NO SEARCH CONDUCTED'),
+                             gr.components.Dropdown(label='Choose the race of the driver', choices=list(tf['OneHotEncoder'].transformers_[0][1].categories_[3]), value='NO SEARCH CONDUCTED'),
                              gr.components.Checkbox(label='Was the driver caught speeding?'),
                              gr.components.Slider(maximum=23, step=1, label='Time HOUR when stop occurred in 24-hour format'),
                              gr.components.Slider(minimum=2012, maximum=2024, step=1, label='Year when stop occurred'),
-                             gr.components.Dropdown(label='What is the name of the subagency that conducted the traffic stop?', choices=tf['OneHotEncoder'].transformers_[0][1].categories_[1], value='4th District, Wheaton'),
+                             gr.components.Dropdown(label='What is the name of the subagency that conducted the traffic stop?', choices=list(tf['OneHotEncoder'].transformers_[0][1].categories_[1]), value='4th District, Wheaton'),
                              gr.components.Checkbox(label='Was the vehicle safe and up to standards (lights properly switched, registration plates attached etc.)?'),
                              gr.components.Slider(minimum=1970, maximum=2023, step=1, label='Year of manufacture of the vehicle:')],
                      outputs=["text"])