Spaces:
Running
Running
add hub search
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ from typing import Optional, Union
|
|
5 |
import gradio as gr
|
6 |
from huggingface_hub import HfApi, Repository
|
7 |
from export import convert
|
8 |
-
|
9 |
|
10 |
|
11 |
DATASET_REPO_URL = "https://huggingface.co/datasets/optimum/exporters"
|
@@ -104,11 +104,11 @@ interface = gr.Interface(
|
|
104 |
max_lines=1,
|
105 |
label="Hugging Face token",
|
106 |
),
|
107 |
-
|
108 |
-
|
109 |
label="Hub Model ID",
|
110 |
-
placeholder="
|
111 |
-
|
112 |
),
|
113 |
],
|
114 |
outputs=[
|
|
|
5 |
import gradio as gr
|
6 |
from huggingface_hub import HfApi, Repository
|
7 |
from export import convert
|
8 |
+
from gradio_huggingfacehub_search import HuggingfaceHubSearch
|
9 |
|
10 |
|
11 |
DATASET_REPO_URL = "https://huggingface.co/datasets/optimum/exporters"
|
|
|
104 |
max_lines=1,
|
105 |
label="Hugging Face token",
|
106 |
),
|
107 |
+
|
108 |
+
HuggingfaceHubSearch(
|
109 |
label="Hub Model ID",
|
110 |
+
placeholder="Search for model id on Huggingface",
|
111 |
+
search_type="model",
|
112 |
),
|
113 |
],
|
114 |
outputs=[
|