YajieXu commited on
Commit
055f598
·
verified ·
1 Parent(s): c798b36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -36,17 +36,12 @@ def download_file_if_any(base_api_url: str, task_id: str) -> str | None:
36
  # --- Basic Agent Definition ---
37
  class BasicAgent:
38
  def __init__(self):
39
- model = HfApiModel(
40
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
41
- max_tokens=2096,
42
- temperature=0.5,
43
- custom_role_conversions=None,
44
- )
45
  self.agent = CodeAgent(
46
  model=model,
47
  tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
48
  add_base_tools=True,
49
- additional_authorized_imports=[]
50
  )
51
  print("BasicAgent initialized.")
52
 
@@ -62,7 +57,7 @@ class BasicAgent:
62
 
63
  # --- Run and Submit ---
64
  def run_and_submit_all(profile: gr.OAuthProfile | None):
65
- space_id = "l3xv/Final_Assignment_Template"
66
  if profile:
67
  username = f"{profile.username}"
68
  else:
 
36
  # --- Basic Agent Definition ---
37
  class BasicAgent:
38
  def __init__(self):
39
+ model=OpenAIServerModel(model_id="gpt-4o"),
 
 
 
 
 
40
  self.agent = CodeAgent(
41
  model=model,
42
  tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
43
  add_base_tools=True,
44
+ additional_authorized_imports=['pandas']
45
  )
46
  print("BasicAgent initialized.")
47
 
 
57
 
58
  # --- Run and Submit ---
59
  def run_and_submit_all(profile: gr.OAuthProfile | None):
60
+ space_id = "YajieXu/Final_Assignment_Template"
61
  if profile:
62
  username = f"{profile.username}"
63
  else: