Using the gradio built in HubSearch component

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
 from huggingface_hub import model_info, hf_hub_url
+from gradio_huggingfacehub_search import HuggingfaceHubSearch
 from netron import build_netron_iframe, is_netron_file
 
 current_repo = None
 
 def repo_change(repo: str, oauth_token: str) -> dict[str, any] | None:
+    print("repo changed")
     global current_repo 
     current_repo = repo
 
@@ -40,12 +42,17 @@ with gr.Blocks() as viewer_app:
         gr.LoginButton(min_width=250)
 
     with gr.Column():
-        repo = gr.Textbox(placeholder="owner/model name (Sign in above to view private models)", label="Model Repository")
+        with gr.Row():
+            model_in = HuggingfaceHubSearch(
+                label="Model ID",
+                placeholder="Search for models on Huggingface",
+                search_type="model",
+            )
         files = gr.Dropdown(choices=[], interactive=False, label="File")
 
     viewer = gr.HTML("""<p>Enter the model ID and select the compatible model file to view the model's graph view.</p>""")
 
-    repo.change(fn=repo_change, inputs=[repo], outputs=[files], trigger_mode='always_last')
+    gr.on([model_in.submit, model_in.change], fn=repo_change, inputs=[model_in], outputs=[files], trigger_mode="always_last")
     files.select(fn=viewer_change, inputs=[files], outputs=[viewer])
 
 if __name__ == "__main__":

requirements.txt

@@ -1 +1,2 @@
 huggingface_hub
+gradio_huggingfacehub_search