fix default indexes

lfoppiano · Jun 22, 2024 · 9ad118d · 9ad118d
1 parent 80dc6af
commit 9ad118d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/streamlit_app.py b/streamlit_app.py
@@ -257,8 +257,7 @@ def play_old_messages():
         "Model:",
         options=OPENAI_MODELS + list(OPEN_MODELS.keys()),
         index=(OPENAI_MODELS + list(OPEN_MODELS.keys())).index(
-            "mistral-7b-instruct-v0.2") if "DEFAULT_MODEL" not in os.environ or not os.environ["DEFAULT_MODEL"] else (
-                OPENAI_MODELS + list(OPEN_MODELS.keys())).index(os.environ["DEFAULT_MODEL"]),
+            os.environ["DEFAULT_MODEL"]) if "DEFAULT_MODEL" not in os.environ or not 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']