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