Spaces:
Running
Running
Niki Zhang
commited on
Update chatbox.py
Browse files- 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-
|
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
|