Hugging Face's logo

Spaces:
mxrkai
/
test24
Sleeping

App Files Community
Niansuh commited on
Commit
c300504
·
verified ·
1 Parent(s): e130770

Update api/config.py

Browse files
Files changed (1) hide show
  1. api/config.py +56 -67
api/config.py CHANGED
@@ -7,81 +7,70 @@ APP_SECRET = os.getenv("APP_SECRET")
7
 
8
  # Allowed models
9
  ALLOWED_MODELS = [
10
- {"id": "chat-gemini-flash", "name": "chat-gemini-flash"},
11
- {"id": "chat-gemini-pro", "name": "chat-gemini-pro"},
12
- {"id": "chat-gpt4m", "name": "chat-gpt4m"},
13
- {"id": "chat-gpt4", "name": "chat-gpt4"},
14
- {"id": "claude-sonnet", "name": "claude-sonnet"},
15
- {"id": "claude-haiku", "name": "claude-haiku"},
16
- {"id": "llama-3-70b", "name": "llama-3-70b"},
17
- {"id": "llama-3-8b", "name": "llama-3-8b"},
18
- {"id": "mistral-large", "name": "mistral-large"},
19
- {"id": "chat-o1-mini", "name": "chat-o1-mini"},
20
- {"id": "flux1", "name": "flux1"},
21
- {"id": "sdxl", "name": "sdxl"},
22
- {"id": "sd", "name": "sd"},
23
- {"id": "sd35", "name": "sd35"},
 
 
 
 
 
24
  ]
25
 
26
  # Model to provider mapping
27
  MODEL_PROVIDER_MAPPING = {
28
- # GizAI models
29
- "chat-gemini-flash": "gizai",
30
- "chat-gemini-pro": "gizai",
31
- "chat-gpt4m": "gizai",
32
- "chat-gpt4": "gizai",
33
- "claude-sonnet": "gizai",
34
- "claude-haiku": "gizai",
35
- "llama-3-70b": "gizai",
36
- "llama-3-8b": "gizai",
37
- "mistral-large": "gizai",
38
- "chat-o1-mini": "gizai",
39
- "flux1": "gizai",
40
- "sdxl": "gizai",
41
- "sd": "gizai",
42
- "sd35": "gizai",
43
  # Aliases
44
- "gemini-flash": "gizai",
45
- "gemini-pro": "gizai",
46
- "gpt-4o-mini": "gizai",
47
- "gpt-4o": "gizai",
48
- "claude-3.5-sonnet": "gizai",
49
- "claude-3-haiku": "gizai",
50
- "llama-3.1-70b": "gizai",
51
- "llama-3.1-8b": "gizai",
52
- "o1-mini": "gizai",
53
- "sd-1.5": "gizai",
54
- "sd-3.5": "gizai",
55
- "flux-schnell": "gizai",
56
  }
57
 
58
  # Model mapping (aliases)
59
  MODEL_MAPPING = {
60
- "chat-gemini-flash": "chat-gemini-flash",
61
- "chat-gemini-pro": "chat-gemini-pro",
62
- "chat-gpt4m": "chat-gpt4m",
63
- "chat-gpt4": "chat-gpt4",
64
- "claude-sonnet": "claude-sonnet",
65
- "claude-haiku": "claude-haiku",
66
- "llama-3-70b": "llama-3-70b",
67
- "llama-3-8b": "llama-3-8b",
68
- "mistral-large": "mistral-large",
69
- "chat-o1-mini": "chat-o1-mini",
70
- "flux1": "flux1",
71
- "sdxl": "sdxl",
72
- "sd": "sd",
73
- "sd35": "sd35",
74
  # Aliases
75
- "gemini-flash": "chat-gemini-flash",
76
- "gemini-pro": "chat-gemini-pro",
77
- "gpt-4o-mini": "chat-gpt4m",
78
- "gpt-4o": "chat-gpt4",
79
- "claude-3.5-sonnet": "claude-sonnet",
80
- "claude-3-haiku": "claude-haiku",
81
- "llama-3.1-70b": "llama-3-70b",
82
- "llama-3.1-8b": "llama-3-8b",
83
- "o1-mini": "chat-o1-mini",
84
- "sd-1.5": "sd",
85
- "sd-3.5": "sd35",
86
- "flux-schnell": "flux1",
87
  }
 
7
 
8
  # Allowed models
9
  ALLOWED_MODELS = [
10
+ {"id": "gpt-4o", "name": "gpt-4o"},
11
+ {"id": "gpt-4o-mini", "name": "gpt-4o-mini"},
12
+ {"id": "o1-preview", "name": "o1-preview"},
13
+ {"id": "o1-mini", "name": "o1-mini"},
14
+ {"id": "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo", "name": "Meta-Llama-3.1-405B"},
15
+ {"id": "meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo", "name": "Llama-3.2-90B-Vision"},
16
+ {"id": "claude-3-sonnet-20240229", "name": "claude-3-sonnet"},
17
+ {"id": "gemini-1.5-pro", "name": "gemini-1.5-pro"},
18
+ {"id": "flux-pro/v1.1", "name": "flux-pro/v1.1"},
19
+ {"id": "flux-realism", "name": "flux-realism"},
20
+ {"id": "flux-pro", "name": "flux-pro"},
21
+ {"id": "dalle-e-3", "name": "dalle-e-3"},
22
+ # Aliases
23
+ {"id": "o1", "name": "o1"},
24
+ {"id": "llama-3.1-405b", "name": "llama-3.1-405b"},
25
+ {"id": "llama-3.2-90b", "name": "llama-3.2-90b"},
26
+ {"id": "claude-3.5-sonnet", "name": "claude-3.5-sonnet"},
27
+ {"id": "gemini-pro", "name": "gemini-pro"},
28
+ {"id": "dalle-3", "name": "dalle-3"},
29
  ]
30
 
31
  # Model to provider mapping
32
  MODEL_PROVIDER_MAPPING = {
33
+ # AmigoChat models
34
+ "gpt-4o": "amigochat",
35
+ "gpt-4o-mini": "amigochat",
36
+ "o1-preview": "amigochat",
37
+ "o1-mini": "amigochat",
38
+ "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo": "amigochat",
39
+ "meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo": "amigochat",
40
+ "claude-3-sonnet-20240229": "amigochat",
41
+ "gemini-1.5-pro": "amigochat",
42
+ "flux-pro/v1.1": "amigochat",
43
+ "flux-realism": "amigochat",
44
+ "flux-pro": "amigochat",
45
+ "dalle-e-3": "amigochat",
 
 
46
  # Aliases
47
+ "o1": "amigochat",
48
+ "llama-3.1-405b": "amigochat",
49
+ "llama-3.2-90b": "amigochat",
50
+ "claude-3.5-sonnet": "amigochat",
51
+ "gemini-pro": "amigochat",
52
+ "dalle-3": "amigochat",
 
 
 
 
 
 
53
  }
54
 
55
  # Model mapping (aliases)
56
  MODEL_MAPPING = {
57
+ "gpt-4o": "gpt-4o",
58
+ "gpt-4o-mini": "gpt-4o-mini",
59
+ "o1-preview": "o1-preview",
60
+ "o1-mini": "o1-mini",
61
+ "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo": "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
62
+ "meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo": "meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo",
63
+ "claude-3-sonnet-20240229": "claude-3-sonnet-20240229",
64
+ "gemini-1.5-pro": "gemini-1.5-pro",
65
+ "flux-pro/v1.1": "flux-pro/v1.1",
66
+ "flux-realism": "flux-realism",
67
+ "flux-pro": "flux-pro",
68
+ "dalle-e-3": "dalle-e-3",
 
 
69
  # Aliases
70
+ "o1": "o1-preview",
71
+ "llama-3.1-405b": "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
72
+ "llama-3.2-90b": "meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo",
73
+ "claude-3.5-sonnet": "claude-3-sonnet-20240229",
74
+ "gemini-pro": "gemini-1.5-pro",
75
+ "dalle-3": "dalle-e-3",
 
 
 
 
 
 
76
  }