Spaces:
Running
Running
Update utils.py
Browse files
utils.py
CHANGED
@@ -8,7 +8,7 @@ from sentence_transformers import SentenceTransformer, CrossEncoder
|
|
8 |
# It formats the output by printing it in color.
|
9 |
|
10 |
# Defined arguments
|
11 |
-
|
12 |
model_name_tr = "gpt-3.5-turbo" # model name
|
13 |
model_name_ta = "gpt-4-1106-preview"
|
14 |
tool_list_path = './tools.json' # list of tools path[ ]
|
@@ -25,7 +25,7 @@ biencoder = SentenceTransformer("BAAI/bge-large-en-v1.5", device="cpu")
|
|
25 |
cross_encoder = CrossEncoder(
|
26 |
"cross-encoder/ms-marco-MiniLM-L-12-v2", max_length=512, device="cpu"
|
27 |
)
|
28 |
-
client = OpenAI(api_key=API_KEY, timeout=60, max_retries=2)
|
29 |
|
30 |
class color:
|
31 |
PURPLE = "\033[1;35;48m"
|
|
|
8 |
# It formats the output by printing it in color.
|
9 |
|
10 |
# Defined arguments
|
11 |
+
|
12 |
model_name_tr = "gpt-3.5-turbo" # model name
|
13 |
model_name_ta = "gpt-4-1106-preview"
|
14 |
tool_list_path = './tools.json' # list of tools path[ ]
|
|
|
25 |
cross_encoder = CrossEncoder(
|
26 |
"cross-encoder/ms-marco-MiniLM-L-12-v2", max_length=512, device="cpu"
|
27 |
)
|
28 |
+
client = OpenAI(api_key=st.secrets["API_KEY"], timeout=60, max_retries=2)
|
29 |
|
30 |
class color:
|
31 |
PURPLE = "\033[1;35;48m"
|