Update main.py
Browse files
main.py
CHANGED
@@ -31,7 +31,7 @@ embeddings = HuggingFaceInferenceAPIEmbeddings(
|
|
31 |
|
32 |
vector_store = SupabaseVectorStore(supabase, embeddings, query_name='match_documents', table_name="documents")
|
33 |
|
34 |
-
models = ["
|
35 |
|
36 |
if openai_api_key:
|
37 |
models += ["gpt-3.5-turbo", "gpt-4"]
|
@@ -77,7 +77,7 @@ if st.session_state["authenticated"]:
|
|
77 |
|
78 |
# Initialize session state variables
|
79 |
if 'model' not in st.session_state:
|
80 |
-
st.session_state['model'] = "
|
81 |
if 'temperature' not in st.session_state:
|
82 |
st.session_state['temperature'] = 0.1
|
83 |
if 'chunk_size' not in st.session_state:
|
|
|
31 |
|
32 |
vector_store = SupabaseVectorStore(supabase, embeddings, query_name='match_documents', table_name="documents")
|
33 |
|
34 |
+
models = ["mistralai/Mixtral-8x7B-Instruct-v0.1", "CohereForAI/c4ai-command-r-plus", "meta-llama/Llama-2-70b-chat-hf"]
|
35 |
|
36 |
if openai_api_key:
|
37 |
models += ["gpt-3.5-turbo", "gpt-4"]
|
|
|
77 |
|
78 |
# Initialize session state variables
|
79 |
if 'model' not in st.session_state:
|
80 |
+
st.session_state['model'] = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
81 |
if 'temperature' not in st.session_state:
|
82 |
st.session_state['temperature'] = 0.1
|
83 |
if 'chunk_size' not in st.session_state:
|