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

Update chain_app.py

Browse files
Files changed (1) hide show
  1. chain_app.py +2 -3
chain_app.py CHANGED
@@ -11,8 +11,7 @@ from fireworks.client import Fireworks
11
 
12
 
13
  hf_token = os.environ.get("HF_TOKEN")
14
- hf_token_llama_3_1 = os.environ.get('HF_TOKEN_FOR_31')
15
- print(hf_token_llama_3_1)
16
  openai_api_key = os.environ.get('OPENAI_API_KEY')
17
  groq_api_key = os.environ.get('GROQ_API_KEY')
18
  cohere_api_key = os.environ.get('COHERE_API_KEY')
@@ -2332,7 +2331,7 @@ async def main(message: cl.Message):
2332
 
2333
  API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3.1-405B"
2334
 
2335
- headers = {"Authorization": "Bearer hf_token_llama_3_1"}
2336
  def query(payload):
2337
  response = requests.post(API_URL, headers=headers, json=payload)
2338
  return response.json()
 
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
 
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()