MohamedRashad commited on
Commit
99871cf
·
1 Parent(s): 7738e98

Update submit_gradio_module to use task_type parameter in model submission

Browse files
Files changed (1) hide show
  1. utils.py +3 -3
utils.py CHANGED
@@ -176,8 +176,8 @@ def load_requests(status_folder):
176
  return df
177
 
178
 
179
- def submit_gradio_module(type):
180
- with gr.Tab(f"Submit {type}") as submitter_tab:
181
  with gr.Row(equal_height=True):
182
  model_name_input = gr.Textbox(
183
  label="Model",
@@ -225,7 +225,7 @@ def submit_gradio_module(type):
225
  )
226
  submit_button.click(
227
  submit_model,
228
- inputs=[model_name_input, revision_input, precision_input, params_input, license_input],
229
  outputs=submission_result
230
  )
231
 
 
176
  return df
177
 
178
 
179
+ def submit_gradio_module(task_type):
180
+ with gr.Tab(f"Submit {task_type}") as submitter_tab:
181
  with gr.Row(equal_height=True):
182
  model_name_input = gr.Textbox(
183
  label="Model",
 
225
  )
226
  submit_button.click(
227
  submit_model,
228
+ inputs=[model_name_input, revision_input, precision_input, params_input, license_input, task_type],
229
  outputs=submission_result
230
  )
231