Niki Zhang commited on
Commit
e48bf0b
·
verified ·
1 Parent(s): 545c152

Update chatbox.py

Browse files
Files changed (1) hide show
  1. chatbox.py +1 -1
chatbox.py CHANGED
@@ -139,7 +139,7 @@ def build_chatbot_tools(load_dict):
139
  class ConversationBot:
140
  def __init__(self, tools, api_key="",save_history=False):
141
  # load_dict = {'VisualQuestionAnswering':'cuda:0', 'ImageCaptioning':'cuda:1',...}
142
- llm = OpenAI(model_name="gpt-3.5-turbo", temperature=0.7, openai_api_key=api_key)
143
  self.llm = llm
144
  self.memory = ConversationBufferMemory(memory_key="chat_history", output_key='output')
145
  self.tools = tools
 
139
  class ConversationBot:
140
  def __init__(self, tools, api_key="",save_history=False):
141
  # load_dict = {'VisualQuestionAnswering':'cuda:0', 'ImageCaptioning':'cuda:1',...}
142
+ llm = OpenAI(model_name="gpt-4o", temperature=0.7, openai_api_key=api_key)
143
  self.llm = llm
144
  self.memory = ConversationBufferMemory(memory_key="chat_history", output_key='output')
145
  self.tools = tools