diff --git a/etl/experiments/ui.py b/etl/experiments/ui.py index 9bf84bf..4e0e0a5 100644 --- a/etl/experiments/ui.py +++ b/etl/experiments/ui.py @@ -38,7 +38,7 @@ vectors = FAISS.from_documents(data, embeddings) vectors.save_local(embedding_pth) -llm = ChatOpenAI(max_tokens=500, temperature=0, model_name="gpt-3.5-turbo") +llm = ChatOpenAI(max_tokens=500, temperature=0, model_name="gpt-3.5-turbo-16k") chain_type_kwargs = {"prompt": CHAT_PROMPT}