Spaces:
Running
Running
Asankhaya Sharma
commited on
Commit
·
0a26e47
1
Parent(s):
7cd26c6
update model names
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 = ["meta-llama/Llama-2-
|
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-
|
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:
|