ecarr-bend commited on
Commit
5ba3489
·
1 Parent(s): abe7873

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -91,9 +91,9 @@ def predict(message, model_type):
91
  # conversational retrieval agent component construction - memory, prompt template, agent, agent executor
92
  # specifying LLM to use
93
  if (model_type==1):
94
- llm = ChatOpenAI(temperature = 0.1, model_name="gpt-3.5-turbo-16k-0613", streaming=True, callbacks=[QueueCallback(q)])
95
  else:
96
- llm = ChatOpenAI(temperature = 0.1, model_name="gpt-4-0613", streaming=True, callbacks=[QueueCallback(q)])
97
 
98
  # This is needed for both the memory and the prompt
99
  memory_key = "history"
 
91
  # conversational retrieval agent component construction - memory, prompt template, agent, agent executor
92
  # specifying LLM to use
93
  if (model_type==1):
94
+ llm = ChatOpenAI(temperature = 0.1, model_name="gpt-3.5-turbo-1106", streaming=True, callbacks=[QueueCallback(q)])
95
  else:
96
+ llm = ChatOpenAI(temperature = 0.1, model_name="gpt-4-1106-preview", streaming=True, callbacks=[QueueCallback(q)])
97
 
98
  # This is needed for both the memory and the prompt
99
  memory_key = "history"