kuzumab commited on
Commit
0702614
·
verified ·
1 Parent(s): cc53000

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,6 +4,7 @@ import requests
4
  import inspect
5
  import pandas as pd
6
  from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, VisitWebpageTool, Tool, HfApiModel, ToolCallingAgent
 
7
  # (Keep Constants as is)
8
  # --- Constants ---
9
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -17,7 +18,7 @@ class AttachmentDownloadTool(Tool):
17
  "description": "task_id that needs to download attachment files."
18
  }
19
  }
20
- output_type = File
21
 
22
  def forward(self, task_id):
23
  download_url = f"{api_url}/files/"
 
4
  import inspect
5
  import pandas as pd
6
  from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, VisitWebpageTool, Tool, HfApiModel, ToolCallingAgent
7
+ import io
8
  # (Keep Constants as is)
9
  # --- Constants ---
10
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
18
  "description": "task_id that needs to download attachment files."
19
  }
20
  }
21
+ output_type = io.BytesIO
22
 
23
  def forward(self, task_id):
24
  download_url = f"{api_url}/files/"