codelion commited on
Commit
3dffd64
·
verified ·
1 Parent(s): 948e819

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 = ["mistralai/Mixtral-8x7B-Instruct-v0.1", "meta-llama/Meta-Llama-3-70B-Instruct", "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'] = "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:
 
31
 
32
  vector_store = SupabaseVectorStore(supabase, embeddings, query_name='match_documents', table_name="documents")
33
 
34
+ models = ["meta-llama/Llama-3.3-70B-Instruct" ,"mistralai/Mixtral-8x7B-Instruct-v0.1", "meta-llama/Meta-Llama-3-70B-Instruct", "CohereForAI/c4ai-command-r-plus"]
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'] = "meta-llama/Llama-3.3-70B-Instruct"
81
  if 'temperature' not in st.session_state:
82
  st.session_state['temperature'] = 0.1
83
  if 'chunk_size' not in st.session_state: