Update app.py

app.py

@@ -131,9 +131,9 @@ def main():
         docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} for i in range(len(texts))]).as_retriever()
         docs = docsearch.get_relevant_documents(query)
         if st.session_state.model_name == 'gpt-4': 
-            max_tkns=8192
+            max_tkns=5500
         else : 
-            max_tkns=4096
+            max_tkns=3000
         chain = load_qa_chain(ChatOpenAI(model = st.session_state.model_name,max_tokens=max_tkns,temperature=0), chain_type="stuff")
         st.session_state.output = chain.run(input_documents=docs, question=query)
         #st.session_state["output"] = output