auto task

app.py

@@ -29,10 +29,11 @@ DEVICE = "cpu"
 LAUNCHER = "process"
 SCENARIO = "inference"
 BACKENDS = ["pytorch", "openvino"]
-MODELS = [
-    "google-bert/bert-base-uncased",
-    "openai-community/gpt2",
-]
+MODELS = ["google-bert/bert-base-uncased", "openai-community/gpt2"]
+MODELS_TO_TASKS = {
+    "google-bert/bert-base-uncased": "text-classification",
+    "openai-community/gpt2": "text-generation",
+}
 TASKS = set(TASKS_TO_OVMODEL.keys()) & set(TASKS_TO_MODEL_LOADERS.keys())
 
 
@@ -157,6 +158,7 @@ def build_demo():
             value="feature-extraction",
             info="Task to run the benchmark on.",
         )
+
         backends = gr.CheckboxGroup(
             interactive=True,
             label="backends",
@@ -168,7 +170,6 @@ def build_demo():
         with gr.Row():
             with gr.Accordion(label="Process Config", open=False, visible=True):
                 process_config = get_process_config()
-
         with gr.Row():
             with gr.Accordion(label="Inference Config", open=False, visible=True):
                 inference_config = get_inference_config()
@@ -179,26 +180,30 @@ def build_demo():
             with gr.Accordion(label="OpenVINO Config", open=False, visible=True):
                 openvino_config = get_openvino_config()
 
-        backends.change(
-            inputs=backends,
-            outputs=backend_configs.children,
-            fn=lambda values: [
-                gr.update(visible=value in values) for value in BACKENDS
-            ],
-        )
-
         with gr.Row():
             button = gr.Button(value="Run Benchmark", variant="primary")
 
-        with gr.Row() as md_output:
+        with gr.Row() as markdown_outputs:
             with gr.Accordion(label="PyTorch Output", open=True, visible=True):
                 pytorch_output = gr.Markdown()
             with gr.Accordion(label="OpenVINO Output", open=True, visible=True):
                 openvino_output = gr.Markdown()
 
+        model.change(
+            inputs=model, outputs=task, fn=lambda value: MODELS_TO_TASKS[value]
+        )
+
+        backends.change(
+            inputs=backends,
+            outputs=backend_configs.children,
+            fn=lambda values: [
+                gr.update(visible=value in values) for value in BACKENDS
+            ],
+        )
+
         backends.change(
             inputs=backends,
-            outputs=md_output.children,
+            outputs=markdown_outputs.children,
             fn=lambda values: [
                 gr.update(visible=value in values) for value in BACKENDS
             ],