fix default model

lfoppiano · Jun 22, 2024 · 3b9ffd5 · 3b9ffd5
1 parent 14963b3
commit 3b9ffd5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/streamlit_app.py b/streamlit_app.py
@@ -254,7 +254,7 @@ def play_old_messages():
         "Model:",
         options=OPENAI_MODELS + list(OPEN_MODELS.keys()),
         index=(OPENAI_MODELS + list(OPEN_MODELS.keys())).index(
-            os.environ["DEFAULT_MODEL"]) if "DEFAULT_MODEL" not in os.environ or not os.environ["DEFAULT_MODEL"] else 0,
+            os.environ["DEFAULT_MODEL"]) if "DEFAULT_MODEL" in os.environ and os.environ["DEFAULT_MODEL"] else 0,
         placeholder="Select model",
         help="Select the LLM model:",
         disabled=st.session_state['doc_id'] is not None or st.session_state['uploaded']