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=[
|