luanpoppe commited on
Commit
7ea334e
·
1 Parent(s): aeb38f0

feat: adicionando token do hugging face

Browse files
Files changed (1) hide show
  1. endpoint_teste/views.py +2 -1
endpoint_teste/views.py CHANGED
@@ -31,8 +31,9 @@ def getTeste(request):
31
  "Resposta": resposta_llm
32
  })
33
  if request.method == "GET":
 
34
  API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B"
35
- headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
36
  def query(payload):
37
  response = requests.post(API_URL, headers=headers, json=payload)
38
  return response.json()
 
31
  "Resposta": resposta_llm
32
  })
33
  if request.method == "GET":
34
+ hugging_face_token = os.environ.get("hugging_face_token")
35
  API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B"
36
+ headers = {"Authorization": "Bearer " + hugging_face_token}
37
  def query(payload):
38
  response = requests.post(API_URL, headers=headers, json=payload)
39
  return response.json()