ayush2917 commited on
Commit
fcc3b47
·
verified ·
1 Parent(s): 8bf6634

Update app/config.py

Browse files
Files changed (1) hide show
  1. 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
- os.getenv("MODEL_A", "meta-llama/Llama-3.1-8B-Instruct").strip(),
12
- "mistralai/Mistral-7B-Instruct-v0.2",
13
- "google/gemma-2b-it",
14
- "HuggingFaceH4/zephyr-7b-beta"
15
  ]
16
 
17
  MODEL_B_LIST = [
18
- os.getenv("MODEL_B", "google/flan-t5-large").strip(),
19
- "google/flan-ul2",
20
- "google/t5-base"
21
  ]
22
 
23
  MODEL_C_LIST = [
24
- os.getenv("MODEL_C", "google/gemma-2b-it").strip(),
25
- "microsoft/phi-2",
26
  "google/flan-t5-large"
27
  ]
28
 
29
  MODEL_D_LIST = [
30
- os.getenv("MODEL_D", "mistralai/Mixtral-8x7B-Instruct-v0.1").strip(),
31
- "meta-llama/Llama-3.1-8B-Instruct",
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