Commit ·
9048350
1
Parent(s): e03fe1c
Update app.py
Browse files
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-
|
| 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 |
|