leannebriffa commited on
Commit
830e6f5
·
1 Parent(s): 1b1cb70
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -99,7 +99,8 @@ def make_prediction(alcohol, arrest_type, belts, contributed_to_accident, disobe
99
 
100
  iface = gr.Interface(fn=make_prediction,
101
  inputs=[gr.components.Checkbox(label='Was the driver under the influence of alcohol?'),
102
- gr.components.Dropdown(label='Choose the arrest type', choices=tf['OneHotEncoder'].transformers_[0][1].categories_[0], value='A - Marked Patrol'),
 
103
  gr.components.Checkbox(label='Were seatbelts used appropriately?'),
104
  gr.components.Checkbox(label='Did the driver actions contribute to an accident?'),
105
  gr.components.Checkbox(label='Was the driver disobedient? (such as failing to display documentation upon request)?'),
 
99
 
100
  iface = gr.Interface(fn=make_prediction,
101
  inputs=[gr.components.Checkbox(label='Was the driver under the influence of alcohol?'),
102
+ #gr.components.Dropdown(label='Choose the arrest type', choices=tf['OneHotEncoder'].transformers_[0][1].categories_[0], value='A - Marked Patrol'),
103
+ 'Marked Patrol',
104
  gr.components.Checkbox(label='Were seatbelts used appropriately?'),
105
  gr.components.Checkbox(label='Did the driver actions contribute to an accident?'),
106
  gr.components.Checkbox(label='Was the driver disobedient? (such as failing to display documentation upon request)?'),