Update app.py

app.py

@@ -137,17 +137,15 @@ def main():
                 max_tkns=3000
             openai.api_key = os.environ["OPENAI_API_KEY"]
             response = openai.ChatCompletion.create(
-                model="gpt-4",
+                model=st.session_state.model_name,
                 messages=[
                     {"role": "system", "content": query},
                     {"role": "user", "content": input_text},
                     ],
                 temperature = 0.7,
-                max_tokens=5500
+                max_tokens=max_tkns
             )
-            st.session_state.output = [
-                choice.message["content"].strip() for choice in response["choices"]
-                ]
+            st.session_state.output = response["choices"][0]["message"]["content"].replace("\\n", "\n")
 #        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