Artin2009 commited on
Commit
2c1b638
·
verified ·
1 Parent(s): 3d7c1d7

Update chain_app.py

Browse files
Files changed (1) hide show
  1. chain_app.py +2 -1
chain_app.py CHANGED
@@ -16,6 +16,7 @@ openai_api_key = os.environ.get('OPENAI_API_KEY')
16
  groq_api_key = os.environ.get('GROQ_API_KEY')
17
  cohere_api_key = os.environ.get('COHERE_API_KEY')
18
  fireworks_api_key = os.environ.get('FIREWORKS_API_KEY')
 
19
 
20
  hf_text_client = Client("Artin2009/text-generation", hf_token=hf_token)
21
  # hf_image_client = Client('Artin2009/image-generation')
@@ -2331,7 +2332,7 @@ async def main(message: cl.Message):
2331
 
2332
  API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3.1-405B"
2333
 
2334
- headers = {"Authorization": "Bearer hf_token_llama_3_1"}
2335
  def query(payload):
2336
  response = requests.post(API_URL, headers=headers, json=payload)
2337
  return response.json()
 
16
  groq_api_key = os.environ.get('GROQ_API_KEY')
17
  cohere_api_key = os.environ.get('COHERE_API_KEY')
18
  fireworks_api_key = os.environ.get('FIREWORKS_API_KEY')
19
+ LLAMA_403_ONLY = os.environ.get('LLAMA_403_ONLY')
20
 
21
  hf_text_client = Client("Artin2009/text-generation", hf_token=hf_token)
22
  # hf_image_client = Client('Artin2009/image-generation')
 
2332
 
2333
  API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3.1-405B"
2334
 
2335
+ headers = {"Authorization": "Bearer LLAMA_403_ONLY"}
2336
  def query(payload):
2337
  response = requests.post(API_URL, headers=headers, json=payload)
2338
  return response.json()