Update app.py
Browse files
app.py
CHANGED
|
@@ -7,13 +7,9 @@ from typing import List, Tuple
|
|
| 7 |
# LLM 모델 정의
|
| 8 |
LLM_MODELS = {
|
| 9 |
"Default": "CohereForAI/c4ai-command-r-plus-08-2024", # 기본 모델
|
| 10 |
-
"
|
| 11 |
-
"
|
| 12 |
-
"
|
| 13 |
-
"Llama2": "meta-llama/Llama-2-7b-chat-hf",
|
| 14 |
-
"Phi": "microsoft/phi-2",
|
| 15 |
-
"Neural": "nvidia/neural-chat-7b-v3-1",
|
| 16 |
-
"Starling": "HuggingFaceH4/starling-lm-7b-alpha"
|
| 17 |
}
|
| 18 |
|
| 19 |
def get_client(model_name):
|
|
|
|
| 7 |
# LLM 모델 정의
|
| 8 |
LLM_MODELS = {
|
| 9 |
"Default": "CohereForAI/c4ai-command-r-plus-08-2024", # 기본 모델
|
| 10 |
+
"Meta": "meta-llama/Llama-3.3-70B-Instruct",
|
| 11 |
+
"Mistral": "mistralai/Mistral-Nemo-Instruct-2407",
|
| 12 |
+
"Alibaba": "Qwen/QwQ-32B-Preview"
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
}
|
| 14 |
|
| 15 |
def get_client(model_name):
|