Spaces:
Running
Running
Update gen_api_answer.py
Browse files- 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 |
|