Update main.py

main.py

@@ -41,15 +41,13 @@ def run_task_model():
     pipe=pipeline(task, model=f'{user}/{model}', **kargs)
     return pipe(input)
 
-
+from huggingface_hub import HfApi
+api=HfApi()
 @app.route("/search")
 def search():
-    response = requests.get(
-        "https://huggingface.co/api/models",
-        params={"limit":20,"full":"True","config":"True", "search": request.args.get("search")},
-        headers={"Authorization":f"Bearer {os.environ.get("token")}"}
-    )
-    return response
+    args=request.args.to_dict()
+    models = api.list_models(**args)
+    return models
 
 @app.route("/task_list")
 def tasks():