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:
|