Spaces:
Sleeping
Sleeping
ddovidovich
commited on
Commit
·
06b6f8b
1
Parent(s):
3d31a3b
Update app.py
Browse files
app.py
CHANGED
@@ -130,8 +130,10 @@ def main():
|
|
130 |
embeddings = OpenAIEmbeddings()
|
131 |
docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} for i in range(len(texts))]).as_retriever()
|
132 |
docs = docsearch.get_relevant_documents(query)
|
133 |
-
|
134 |
-
|
|
|
|
|
135 |
chain = load_qa_chain(ChatOpenAI(model = st.session_state.model_name,max_tokens=max_tkns,temperature=0), chain_type="stuff")
|
136 |
st.session_state.output = chain.run(input_documents=docs, question=query)
|
137 |
#st.session_state["output"] = output
|
|
|
130 |
embeddings = OpenAIEmbeddings()
|
131 |
docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} for i in range(len(texts))]).as_retriever()
|
132 |
docs = docsearch.get_relevant_documents(query)
|
133 |
+
if st.session_state.model_name == 'gpt-4':
|
134 |
+
max_tkns=8192
|
135 |
+
else :
|
136 |
+
max_tkns=4096
|
137 |
chain = load_qa_chain(ChatOpenAI(model = st.session_state.model_name,max_tokens=max_tkns,temperature=0), chain_type="stuff")
|
138 |
st.session_state.output = chain.run(input_documents=docs, question=query)
|
139 |
#st.session_state["output"] = output
|