Update app/config.py
Browse files- app/config.py +11 -13
app/config.py
CHANGED
|
@@ -6,32 +6,30 @@ load_dotenv()
|
|
| 6 |
|
| 7 |
HF_TOKEN = os.getenv("HF_TOKEN", "")
|
| 8 |
|
| 9 |
-
# Primary → fallback models
|
| 10 |
MODEL_A_LIST = [
|
| 11 |
-
|
| 12 |
-
"
|
| 13 |
-
"
|
| 14 |
-
"
|
| 15 |
]
|
| 16 |
|
| 17 |
MODEL_B_LIST = [
|
| 18 |
-
|
| 19 |
-
"google/flan-
|
| 20 |
-
"
|
| 21 |
]
|
| 22 |
|
| 23 |
MODEL_C_LIST = [
|
| 24 |
-
|
| 25 |
-
"microsoft/phi-2",
|
| 26 |
"google/flan-t5-large"
|
| 27 |
]
|
| 28 |
|
| 29 |
MODEL_D_LIST = [
|
| 30 |
-
|
| 31 |
-
"
|
| 32 |
-
"HuggingFaceH4/zephyr-7b-beta"
|
| 33 |
]
|
| 34 |
|
|
|
|
| 35 |
SECRET_KEY = os.getenv("SECRET_KEY", "change_this_secret")
|
| 36 |
ACCESS_TOKEN_EXPIRE_MINUTES = int(os.getenv("ACCESS_TOKEN_EXPIRE_MINUTES", "120"))
|
| 37 |
|
|
|
|
| 6 |
|
| 7 |
HF_TOKEN = os.getenv("HF_TOKEN", "")
|
| 8 |
|
|
|
|
| 9 |
MODEL_A_LIST = [
|
| 10 |
+
"tiiuae/falcon-7b-instruct",
|
| 11 |
+
"HuggingFaceH4/zephyr-7b-alpha",
|
| 12 |
+
"EleutherAI/pythia-2.8b-deduped",
|
| 13 |
+
"google/flan-t5-base"
|
| 14 |
]
|
| 15 |
|
| 16 |
MODEL_B_LIST = [
|
| 17 |
+
"google/flan-t5-large",
|
| 18 |
+
"google/flan-t5-base",
|
| 19 |
+
"tiiuae/falcon-3b-instruct"
|
| 20 |
]
|
| 21 |
|
| 22 |
MODEL_C_LIST = [
|
| 23 |
+
"tiiuae/falcon-7b-instruct",
|
|
|
|
| 24 |
"google/flan-t5-large"
|
| 25 |
]
|
| 26 |
|
| 27 |
MODEL_D_LIST = [
|
| 28 |
+
"tiiuae/falcon-7b-instruct",
|
| 29 |
+
"HuggingFaceH4/zephyr-7b-alpha"
|
|
|
|
| 30 |
]
|
| 31 |
|
| 32 |
+
|
| 33 |
SECRET_KEY = os.getenv("SECRET_KEY", "change_this_secret")
|
| 34 |
ACCESS_TOKEN_EXPIRE_MINUTES = int(os.getenv("ACCESS_TOKEN_EXPIRE_MINUTES", "120"))
|
| 35 |
|