Spaces:
Sleeping
Sleeping
cd@bziiit.com 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(".")
|