Spaces:
Running
Running
Commit
·
fe9941d
1
Parent(s):
a81a191
remove nvidia nim
Browse files
app.py
CHANGED
@@ -58,28 +58,18 @@ CONGRESS_GOV_TYPE_MAP = {
|
|
58 |
}
|
59 |
OPENAI_CHAT_MODELS = [
|
60 |
"gpt-4o-mini",
|
61 |
-
"gpt-3.5-turbo-0125",
|
62 |
"gpt-4o",
|
63 |
]
|
64 |
ANTHROPIC_CHAT_MODELS = [
|
65 |
-
"claude-3-opus-20240229",
|
66 |
-
"claude-3-sonnet-20240229",
|
67 |
-
"claude-3-5-sonnet-20240620",
|
68 |
"claude-3-haiku-20240307",
|
|
|
|
|
69 |
]
|
70 |
-
|
71 |
-
"microsoft/phi-3-mini-128k-instruct",
|
72 |
-
"google/gemma-7b",
|
73 |
-
"meta/llama3-8b-instruct",
|
74 |
-
"meta/llama3-70b-instruct",
|
75 |
-
"mistralai/mixtral-8x22b-instruct-v0.1",
|
76 |
-
]
|
77 |
-
CHAT_MODELS = OPENAI_CHAT_MODELS + ANTHROPIC_CHAT_MODELS + NVIDIA_NIM_CHAT_MODELS
|
78 |
|
79 |
PROVIDER_MODELS = {
|
80 |
"OpenAI": OPENAI_CHAT_MODELS,
|
81 |
"Anthropic": ANTHROPIC_CHAT_MODELS,
|
82 |
-
"Nvidia NIM": NVIDIA_NIM_CHAT_MODELS,
|
83 |
}
|
84 |
|
85 |
|
|
|
58 |
}
|
59 |
OPENAI_CHAT_MODELS = [
|
60 |
"gpt-4o-mini",
|
|
|
61 |
"gpt-4o",
|
62 |
]
|
63 |
ANTHROPIC_CHAT_MODELS = [
|
|
|
|
|
|
|
64 |
"claude-3-haiku-20240307",
|
65 |
+
"claude-3-5-sonnet-20240620",
|
66 |
+
"claude-3-opus-20240229",
|
67 |
]
|
68 |
+
CHAT_MODELS = OPENAI_CHAT_MODELS + ANTHROPIC_CHAT_MODELS
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
69 |
|
70 |
PROVIDER_MODELS = {
|
71 |
"OpenAI": OPENAI_CHAT_MODELS,
|
72 |
"Anthropic": ANTHROPIC_CHAT_MODELS,
|
|
|
73 |
}
|
74 |
|
75 |
|