Update multi_inference.py
Browse files- multi_inference.py +12 -10
multi_inference.py
CHANGED
@@ -3,20 +3,22 @@
|
|
3 |
import requests
|
4 |
import os
|
5 |
|
6 |
-
#
|
7 |
DEEPSEEK_KEYS = [
|
8 |
-
"
|
9 |
-
"
|
10 |
-
"
|
11 |
-
"
|
|
|
12 |
]
|
13 |
|
|
|
14 |
OPENAI_KEYS = [
|
15 |
-
"
|
16 |
-
"
|
17 |
-
"
|
18 |
-
"
|
19 |
-
"
|
20 |
]
|
21 |
|
22 |
|
|
|
3 |
import requests
|
4 |
import os
|
5 |
|
6 |
+
# Load DeepSeek keys from Hugging Face Secrets
|
7 |
DEEPSEEK_KEYS = [
|
8 |
+
os.getenv("DEEPSEEK_KEY_1", "").strip(),
|
9 |
+
os.getenv("DEEPSEEK_KEY_2", "").strip(),
|
10 |
+
os.getenv("DEEPSEEK_KEY_3", "").strip(),
|
11 |
+
os.getenv("DEEPSEEK_KEY_4", "").strip(),
|
12 |
+
os.getenv("DEEPSEEK_KEY_5", "").strip()
|
13 |
]
|
14 |
|
15 |
+
# Load OpenAI keys named as open1, open2, etc.
|
16 |
OPENAI_KEYS = [
|
17 |
+
os.getenv("open1", "").strip(),
|
18 |
+
os.getenv("open2", "").strip(),
|
19 |
+
os.getenv("open3", "").strip(),
|
20 |
+
os.getenv("open4", "").strip(),
|
21 |
+
os.getenv("open5", "").strip()
|
22 |
]
|
23 |
|
24 |
|