Cartinoe5930 commited on
Commit
b9eb2d7
1 Parent(s): c8005a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -54,6 +54,7 @@ def warmup(model_list=warmup_test, model_inference_endpoints=inference_endpoint)
54
  for i in range(len(model_list)):
55
  API_URL = model_inference_endpoints[model_list[i]["API_URL"]]
56
  headers = model_inference_endpoints[model_list[i]["headers"]]
 
57
 
58
  def query(payload):
59
  response = requests.post(API_URL, headers=headers, json=payload)
 
54
  for i in range(len(model_list)):
55
  API_URL = model_inference_endpoints[model_list[i]["API_URL"]]
56
  headers = model_inference_endpoints[model_list[i]["headers"]]
57
+ headers["Authorization"] += HF_TOKEN
58
 
59
  def query(payload):
60
  response = requests.post(API_URL, headers=headers, json=payload)