ddovidovich commited on
Commit
06b6f8b
·
1 Parent(s): 3d31a3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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
- if st.session_state.model_name == 'gpt-4': max_tkns=8192
134
- else : max_tkns=4096
 
 
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