kaikaidai commited on
Commit
4ae4a5a
·
verified ·
1 Parent(s): 4959421

Update gen_api_answer.py

Browse files
Files changed (1) hide show
  1. gen_api_answer.py +2 -0
gen_api_answer.py CHANGED
@@ -15,12 +15,14 @@ from prompts import (
15
  FLOW_JUDGE_PROMPT
16
  )
17
  from transformers import AutoTokenizer
 
18
 
19
  # Initialize clients
20
  anthropic_client = anthropic.Anthropic()
21
  openai_client = OpenAI()
22
  together_client = Together()
23
  hf_api_key = os.getenv("HF_API_KEY")
 
24
  flow_judge_api_key = os.getenv("FLOW_JUDGE_API_KEY")
25
  cohere_client = cohere.ClientV2(os.getenv("CO_API_KEY"))
26
 
 
15
  FLOW_JUDGE_PROMPT
16
  )
17
  from transformers import AutoTokenizer
18
+ from huggingface_hub import login
19
 
20
  # Initialize clients
21
  anthropic_client = anthropic.Anthropic()
22
  openai_client = OpenAI()
23
  together_client = Together()
24
  hf_api_key = os.getenv("HF_API_KEY")
25
+ login(hf_api_key)
26
  flow_judge_api_key = os.getenv("FLOW_JUDGE_API_KEY")
27
  cohere_client = cohere.ClientV2(os.getenv("CO_API_KEY"))
28