echarlaix HF staff commited on
Commit
15c5870
1 Parent(s): 583caa1

add hub search

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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
- # from gradio_huggingfacehub_search import HuggingfaceHubSearch
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
- # HuggingfaceHubSearch(
108
- gr.Textbox(
109
  label="Hub Model ID",
110
- placeholder="distilbert/distilbert-base-cased-distilled-squad",
111
- # search_type="model",
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=[