Update model_selector.py
Browse files- model_selector.py +5 -4
model_selector.py
CHANGED
@@ -3,9 +3,10 @@ from transformers import AutoModelForSequenceClassification
|
|
3 |
def get_model_list():
|
4 |
# List of popular pre-trained models
|
5 |
models = [
|
6 |
-
"
|
7 |
-
"
|
8 |
-
"
|
9 |
-
"
|
|
|
10 |
]
|
11 |
return models
|
|
|
3 |
def get_model_list():
|
4 |
# List of popular pre-trained models
|
5 |
models = [
|
6 |
+
"Qwen/Qwen2.5-Coder-32B-Instruct",
|
7 |
+
"microsoft/Phi-3-mini-4k-instruct",
|
8 |
+
"bigcode/starcoder2-3b",
|
9 |
+
"codellama/CodeLlama-34b-Instruct-hf"
|
10 |
+
"microsoft/Phi-3.5-mini-instruct"
|
11 |
]
|
12 |
return models
|