[email protected] commited on
Commit
08069bc
·
1 Parent(s): ada188d

Set default model to gpt-4o

Browse files
Files changed (1) hide show
  1. model/selector.py +6 -1
model/selector.py CHANGED
@@ -34,8 +34,13 @@ def ModelSelector():
34
  model_options.append(option_name)
35
  model_mapping[option_name] = model_instance
36
 
 
 
 
 
 
37
  # Sélection d'un modèle via un seul sélecteur
38
- selected_model_option = st.selectbox("Choisissez votre modèle", options=model_options)
39
 
40
  if(st.session_state["assistant"]):
41
  splitter = model_mapping[selected_model_option].split(".")
 
34
  model_options.append(option_name)
35
  model_mapping[option_name] = model_instance
36
 
37
+ default_model = "OpenAI - gpt-4o"
38
+
39
+ # Trouver l'index du modèle par défaut
40
+ default_index = model_options.index(default_model) if default_model in model_options else 0
41
+
42
  # Sélection d'un modèle via un seul sélecteur
43
+ selected_model_option = st.selectbox("Choisissez votre modèle", options=model_options, index=default_index)
44
 
45
  if(st.session_state["assistant"]):
46
  splitter = model_mapping[selected_model_option].split(".")