add openvino model

app.py

@@ -46,6 +46,8 @@ def parse_configs(inputs):
             model = value
         elif key.label == "task":
             task = value
+        elif key.label == "openvino_model":
+            openvino_label = value
         elif "." in key.label:
             backend, argument = key.label.split(".")
             configs[backend][argument] = value
@@ -65,9 +67,10 @@ def parse_configs(inputs):
         device=DEVICE,
         **configs["pytorch"],
     )
+
     configs["openvino"] = OVConfig(
         task=task,
-        model=model,
+        model=openvino_label or model,
         device=DEVICE,
         **configs["openvino"],
     )
@@ -165,12 +168,17 @@ with gr.Blocks() as demo:
 
     with gr.Column(variant="panel"):
         model = HuggingfaceHubSearch(
-            placeholder="Search for a model",
-            sumbit_on_select=True,
+            placeholder="Search for a PyTorch model",
             search_type="model",
             label="model",
         )
 
+        openvino_model = HuggingfaceHubSearch(
+            placeholder="Search for an OpenVINO model (optional)",
+            search_type="model",
+            label="openvino_model",
+        )
+
         with gr.Row():
             task = gr.Dropdown(
                 info="Task to run the benchmark on.",
@@ -207,6 +215,7 @@ with gr.Blocks() as demo:
         inputs={
             task,
             model,
+            openvino_model,
             # backends,
             *process_config.values(),
             *inference_config.values(),

config_store.py

@@ -73,21 +73,11 @@ def get_pytorch_config():
             choices=["bfloat16", "float16", "float32", "auto"],
             info="The dtype to use for the model",
         ),
-        "pytorch.torch_compile": gr.Checkbox(
-            value=False,
-            label="pytorch.torch_compile",
-            info="Compiles the model with torch.compile",
-        ),
     }
 
 
 def get_openvino_config():
     return {
-        "openvino.export": gr.Checkbox(
-            value=True,
-            label="openvino.export",
-            info="Exports the model to OpenVINO",
-        ),
         "openvino.use_cache": gr.Checkbox(
             value=True,
             label="openvino.use_cache",