Diana1234 commited on
Commit
ef0b16f
1 Parent(s): e3f7cdd

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
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
- API_KEY = st.write(st.secrets["API_KEY"])# api_key
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"