MohamedRashad commited on
Commit
9d57dd4
·
1 Parent(s): 99871cf

Add Levenshtein dependency and update task_type handling in submit_gradio_module

Browse files
Files changed (2) hide show
  1. requirements.txt +2 -1
  2. utils.py +1 -1
requirements.txt CHANGED
@@ -1 +1,2 @@
1
- fuzzywuzzy
 
 
1
+ fuzzywuzzy
2
+ Levenshtein
utils.py CHANGED
@@ -225,7 +225,7 @@ def submit_gradio_module(task_type):
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
 
 
225
  )
226
  submit_button.click(
227
  submit_model,
228
+ inputs=[model_name_input, revision_input, precision_input, params_input, license_input, gr.State(value=task_type)],
229
  outputs=submission_result
230
  )
231