Canstralian commited on
Commit
4a6742e
·
verified ·
1 Parent(s): 68421f4

Update model_selector.py

Browse files
Files changed (1) hide show
  1. 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
- "bert-base-uncased",
7
- "distilbert-base-uncased",
8
- "roberta-base",
9
- "gpt2"
 
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