bstraehle commited on
Commit
2b3b091
·
verified ·
1 Parent(s): 74fdc64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,7 +3,8 @@ import os
3
 
4
  from multi_agent import run_multi_agent
5
 
6
- LLM = "gpt-4o"
 
7
 
8
  def invoke(openai_api_key, num_turns = 2):
9
  if (openai_api_key == ""):
@@ -11,7 +12,7 @@ def invoke(openai_api_key, num_turns = 2):
11
 
12
  os.environ["OPENAI_API_KEY"] = openai_api_key
13
 
14
- return run_multi_agent(LLM, num_turns)
15
 
16
  gr.close_all()
17
 
 
3
 
4
  from multi_agent import run_multi_agent
5
 
6
+ LLM_WHITE = "gpt-3.5-turbo"
7
+ LLM_BLACK = "gpt-4o"
8
 
9
  def invoke(openai_api_key, num_turns = 2):
10
  if (openai_api_key == ""):
 
12
 
13
  os.environ["OPENAI_API_KEY"] = openai_api_key
14
 
15
+ return run_multi_agent(LLM_WHITE, LLM_BLACK, num_turns)
16
 
17
  gr.close_all()
18