fix interface

app.py

@@ -5,6 +5,7 @@ from typing import Optional, Union
 import gradio as gr
 from huggingface_hub import HfApi, Repository
 from export import convert
+from gradio_huggingfacehub_search import HuggingfaceHubSearch
 
 
 DATASET_REPO_URL = "https://huggingface.co/datasets/optimum/exporters"
@@ -88,7 +89,7 @@ This Space uses [Optimum Intel](https://huggingface.co/docs/optimum/intel/infere
 
 To export your model you need:
 - A read-access token from [https://huggingface.co/settings/tokens](https://huggingface.co/settings/tokens).
-- A  model id from the Hub (for example: [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english))
+- A Model ID from the Hub (for example: [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english))
 
 
 That's it ! 🔥
@@ -96,33 +97,30 @@ That's it ! 🔥
 After the model conversion, we will open a PR against the source repo.
 """
 
+interface = gr.Interface(
+    fn=export,
+    inputs=[
+        HuggingfaceHubSearch(
+            label="Hub Model ID",
+            placeholder="Search for model id on Huggingface",
+            search_type="model",
+        ),
+        gr.Textbox(
+            max_lines=1,
+            label="Hugging Face token",
+        ),
+    ],
+    outputs=[
+        gr.Markdown(label="output"),
+        gr.Image(show_label=False),
+    ],
+    submit_btn=gr.Button("Export"),
+    title=TITLE,
+    description=DESCRIPTION,
+)
 with gr.Blocks() as demo:
     gr.HTML(TTILE_IMAGE)
     gr.HTML(TITLE)
-
-    with gr.Row():
-        with gr.Column(scale=1):
-            gr.Markdown(DESCRIPTION)
-
-        with gr.Column(scale=1):
-            input_token = gr.Textbox(
-                max_lines=1,
-                label="Hugging Face token",
-            )
-            input_model = gr.Textbox(
-                max_lines=1,
-                label="Model name",
-                placeholder="distilbert-base-uncased-finetuned-sst-2-english",
-            )
-
-            btn = gr.Button("Export")
-            output = gr.Markdown(label="Output")
-
-    btn.click(
-        fn=export,
-        inputs=[input_token, input_model],
-        outputs=output,
-    )
-
+    interface.render()
 
 demo.launch()