atrytone commited on
Commit
5f9d4f4
1 Parent(s): 5c7bf4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -142,19 +142,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
142
  visible=False
143
  )
144
 
145
- action_btn1.click(fn=lambda x: inference(x, 1),
146
  inputs=[
147
  abst,
148
  ],
149
  outputs=[a_output, j_output, n_output],
150
  api_name="neurojane")
151
- action_btn2.click(fn=lambda x: inference(x, 2),
152
  inputs=[
153
  abst,
154
  ],
155
  outputs=[a_output, j_output, n_output],
156
  api_name="neurojane")
157
- action_btn3.click(fn=lambda x: inference(x, 3),
158
  inputs=[
159
  abst,
160
  ],
 
142
  visible=False
143
  )
144
 
145
+ action_btn1.click(fn=lambda x: inference(x, index_names[0]),
146
  inputs=[
147
  abst,
148
  ],
149
  outputs=[a_output, j_output, n_output],
150
  api_name="neurojane")
151
+ action_btn2.click(fn=lambda x: inference(x, index_names[1]),
152
  inputs=[
153
  abst,
154
  ],
155
  outputs=[a_output, j_output, n_output],
156
  api_name="neurojane")
157
+ action_btn3.click(fn=lambda x: inference(x, index_names[2]),
158
  inputs=[
159
  abst,
160
  ],