Update app.py

app.py

@@ -2,12 +2,14 @@ import gradio as gr
 import json
 from huggingface_hub import HfApi, ModelFilter, list_models
 api = HfApi()
-def model_explorer(limit=100,task='text-generation'):
+def model_explorer(search, limit=100,task='text-generation'):
     # List all models
+    if not search:
+        search=None
     out_box=[]
     filt = ModelFilter(task="text-generation")
     
-    this = api.list_models(limit=limit,filter=filt,cardData=True)
+    this = api.list_models(search=search,limit=limit,filter=filt,cardData=True)
     
     for i,mod in enumerate(this):
         out_box.append(mod)
@@ -80,8 +82,10 @@ def fix_json():
         if line.strip():
             print (line)
 with gr.Blocks() as app:
-    limit=gr.Slider(minimum=1,maximum=100,value=100)
+    with gr.Row():
+        search=gr.Textbox()
+        limit=gr.Slider(minimum=1,maximum=100,value=100)
     btn=gr.Button()
     models_json=gr.JSON()
-    btn.click(model_explorer,[limit],models_json)
+    btn.click(model_explorer,[search,limit],models_json)
 app.launch()