peichao.dong commited on
Commit
9bcc3e3
·
1 Parent(s): e5580aa

update openai model to 3.5-turbo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from langchain.memory.prompt import ENTITY_MEMORY_CONVERSATION_TEMPLATE
8
 
9
  openai.api_key = os.environ.get("OPENAI_API_KEY")
10
 
11
- llm = OpenAI(model_name="text-davinci-003", stop="\n\n",
12
  max_tokens=2048, temperature=0.5)
13
  entityMemory = ConversationEntityMemory(llm=llm)
14
 
 
8
 
9
  openai.api_key = os.environ.get("OPENAI_API_KEY")
10
 
11
+ llm = OpenAI(model_name="gpt-3.5-turbo", stop="\n\n",
12
  max_tokens=2048, temperature=0.5)
13
  entityMemory = ConversationEntityMemory(llm=llm)
14