ecarr-bend commited on
Commit
9048350
·
1 Parent(s): e03fe1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -91,7 +91,7 @@ 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-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
 
 
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", 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