Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import json
 from huggingface_hub import HfApi, ModelFilter, list_models
 api = HfApi()
-def model_explorer(search,limit=100,task='text-generation'):
+def model_explorer(search,task,limit=100):
     # List all models
     if limit==0:
         limit=None
@@ -89,10 +89,11 @@ def fix_json():
             print (line)
 with gr.Blocks() as app:
     with gr.Row():
-        search=gr.Textbox()
+        search=gr.Textbox(label="query")
+        task=gr.Textbox(label="task")
         limit=gr.Slider(minimum=0,maximum=10000,value=100)
     btn=gr.Button()
     names_json=gr.JSON()
     models_json=gr.JSON()
-    btn.click(model_explorer,[search,limit],[models_json,names_json])
+    btn.click(model_explorer,[search,task,limit],[models_json,names_json])
 app.launch()