Spaces:
Running
Running
[email protected]
commited on
Commit
·
08069bc
1
Parent(s):
ada188d
Set default model to gpt-4o
Browse files- 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(".")
|