codelion commited on
Commit
4546e02
1 Parent(s): fc0b9e9

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
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 = ["meta-llama/Llama-2-70b-chat-hf", "mistralai/Mixtral-8x7B-Instruct-v0.1", "CohereForAI/c4ai-command-r-plus"]
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'] = "meta-llama/Llama-2-70b-chat-hf"
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: