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 |
|