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

hugging face api key request support

Browse Source
This commit is contained in:
dawoodkhan82 2021-05-07 05:17:26 -04:00
parent 1c14d2d2bd
commit 81c33b51e8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gradio/external.py
View File

@ -8,7 +8,7 @@ def get_huggingface_interface(model_name, api_key):
headers = {"Authorization": f"Bearer {api_key}"}
# Checking if model exists, and if so, it gets the pipeline
response = requests.request("GET", api_url)
response = requests.request("GET", api_url, headers=headers)
assert response.status_code == 200, "Invalid model name or src"
p = response.json().get('pipeline_tag')
@ -80,7 +80,7 @@ def get_huggingface_interface(model_name, api_key):
payload = pipeline['preprocess'](*input)
payload.update({'options': {'wait_for_model': True}})
data = json.dumps(payload)
response = requests.request("POST", api_url, data=data)
response = requests.request("POST", api_url, headers=headers, data=data)
result = json.loads(response.content.decode("utf-8"))
output = pipeline['postprocess'](result)
return output