Update utils.py to add 'Task' column in DataFrame and refactor submit_gradio_module to use state variable for task_type

utils.py

@@ -16,7 +16,7 @@ def load_retrieval_results():
 
     retrieval_dataframe_path = results_dir / "retrieval_results.json"
     if not retrieval_dataframe_path.exists():
-        df = pd.DataFrame(columns=["Model", 'Revision', 'Precision', "Model Size", "Embedding Dimension", "Max Tokens", "Web Search Dataset (MRR)", "Web Search Dataset (nDCG@k=None)"])
+        df = pd.DataFrame(columns=["Model", "Revision", "Precision", "Task", "Model Size", "Embedding Dimension", "Max Tokens", "Web Search Dataset (MRR)", "Web Search Dataset (nDCG@k=None)"])
     else:
         df = pd.read_json(retrieval_dataframe_path)
     return df
@@ -177,6 +177,7 @@ def load_requests(status_folder):
 
 
 def submit_gradio_module(task_type):
+    var = gr.State(value=task_type)
     with gr.Tab(f"Submit {task_type}") as submitter_tab: 
         with gr.Row(equal_height=True):
             model_name_input = gr.Textbox(
@@ -225,7 +226,7 @@ def submit_gradio_module(task_type):
         )
         submit_button.click(
             submit_model,
-            inputs=[model_name_input, revision_input, precision_input, params_input, license_input, gr.State(value=task_type)],
+            inputs=[model_name_input, revision_input, precision_input, params_input, license_input, var],
             outputs=submission_result
         )