Asankhaya Sharma commited on
Commit
0a26e47
·
1 Parent(s): 7cd26c6

update model names

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-7b-chat-hf", "mistralai/Mixtral-8x7B-Instruct-v0.1"]
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-7b-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 = ["meta-llama/Llama-2-70b-chat-hf", "mistralai/Mixtral-8x7B-Instruct-v0.1"]
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-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: