Fixed error in app.py

app.py

@@ -96,24 +96,14 @@ def make_prediction(alcohol, arrest_type, belts, contributed_to_accident, disobe
     # Return the prediction
     return prediction[0]
 
-arrest_types = ['A - Marked Patrol', 'G - Marked Moving Radar (Stationary)',
-       'Q - Marked Laser', 'L - Motorcycle',
-       'H - Unmarked Moving Radar (Stationary)', 'O - Foot Patrol',
-       'E - Marked Stationary Radar', 'B - Unmarked Patrol',
-       'S - License Plate Recognition', 'R - Unmarked Laser',
-       'J - Unmarked Moving Radar (Moving)', 'M - Marked (Off-Duty)',
-       'I - Marked Moving Radar (Moving)',
-       'F - Unmarked Stationary Radar', 'D - Unmarked VASCAR',
-       'K - Aircraft Assist', 'C - Marked VASCAR', 'P - Mounted Patrol',
-       'N - Unmarked (Off-Duty)']
 
 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=arrest_types, value='A - Marked Patrol'),
+                             gr.components.Dropdown(label='Choose the arrest type', choices=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)?'),
-                             gr.components.Dropdown(label='Choose the arrest type', choices=tf['OrdinalEncoder_VeryHighCardinality'].categories_, value='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'),