fix

app.py

@@ -116,7 +116,6 @@ interface = gr.Interface(
     ],
     outputs=[
         gr.Markdown(label="output"),
-        gr.Image(show_label=False),
     ],
     submit_btn=gr.Button("Export"),
     title=TITLE,

export.py

@@ -30,6 +30,7 @@ from optimum.intel.openvino import (
 )
 from optimum.intel.utils.constant import _TASK_ALIASES
 from optimum.intel.openvino.utils import _HEAD_TO_AUTOMODELS
+from optimum.exporters import TasksManager
 
 SPACES_URL = "https://huggingface.co/spaces/echarlaix/openvino-export"
 
@@ -49,6 +50,15 @@ def previous_pr(api: "HfApi", model_id: str, pr_title: str) -> Optional["Discuss
 
 
 def convert_openvino(model_id: str, task: str, folder: str) -> List:
+    task = TasksManager.map_from_synonym(task)
+    if task == "auto":
+        try:
+            task = TasksManager.infer_task_from_model(model_id)
+        except KeyError as e:
+            raise KeyError(
+                    f"The task could not be automatically inferred. Please provide the task argument with the relevant task from {', '.join(TasksManager.get_all_tasks())}.  {e}"
+            )
+
     task = _TASK_ALIASES.get(task, task)
     if task not in _HEAD_TO_AUTOMODELS:
         raise ValueError(f"The task '{task}' is not supported, only {_HEAD_TO_AUTOMODELS.keys()} tasks are supported")