mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2024-12-27 02:30:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

huggingface api key setup

Browse Source
This commit is contained in:
dawoodkhan82 2021-05-07 04:29:27 -04:00
parent afd55beb3b
commit 2c83ef9291
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gradio/external.py
View File

@ -6,7 +6,8 @@ from gradio import inputs, outputs
def get_huggingface_interface(model_name, api_key):
api_url = "https://api-inference.huggingface.co/models/{}".format(model_name)
headers = {"Authorization": f"Bearer {api_key}"}
# Checking if model exists, and if so, it gets the pipeline
response = requests.request("GET", api_url)
assert response.status_code == 200, "Invalid model name or src"