Artin2009 commited on
Commit
3d7c1d7
1 Parent(s): 08001a1

Update chain_app.py

Browse files
Files changed (1) hide show
  1. chain_app.py +2 -2
chain_app.py CHANGED
@@ -11,7 +11,7 @@ from fireworks.client import Fireworks
11
 
12
 
13
  hf_token = os.environ.get("HF_TOKEN")
14
- llama405_token = os.environ.get('HF_TOKEN_FOR_31')
15
  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')
@@ -2331,7 +2331,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 llama405_token"}
2335
  def query(payload):
2336
  response = requests.post(API_URL, headers=headers, json=payload)
2337
  return response.json()
 
11
 
12
 
13
  hf_token = os.environ.get("HF_TOKEN")
14
+ hf_token_llama_3_1 = os.environ.get('HF_TOKEN_FOR_31')
15
  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')
 
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()