tournas commited on
Commit
785aaa0
·
verified ·
1 Parent(s): 288c481

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,12 +2,13 @@ import datetime
2
  import requests
3
  import pytz
4
  import yaml
 
5
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
6
  from tools.final_answer import FinalAnswerTool
7
  from Gradio_UI import GradioUI
8
 
9
  # Define a tool to retrieve the Discounted Cash Flow (DCF) of a company
10
-
11
  @tool
12
  def get_dcf_of_company(api_key: str, company_name: str) -> str:
13
  """A tool that retrieves the DCF of a company based on its name.
 
2
  import requests
3
  import pytz
4
  import yaml
5
+ import spaces
6
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
7
  from tools.final_answer import FinalAnswerTool
8
  from Gradio_UI import GradioUI
9
 
10
  # Define a tool to retrieve the Discounted Cash Flow (DCF) of a company
11
+ @spaces.GPU
12
  @tool
13
  def get_dcf_of_company(api_key: str, company_name: str) -> str:
14
  """A tool that retrieves the DCF of a company based on its name.