ZennyKenny commited on
Commit
4943ed5
·
verified ·
1 Parent(s): 4f04c00

remove GPU decorator

Browse files
Files changed (1) hide show
  1. app.py +0 -7
app.py CHANGED
@@ -5,15 +5,11 @@ from sqlalchemy import text
5
  from smolagents import tool, CodeAgent, HfApiModel
6
  import spaces
7
 
8
- # Import the database setup from database.py
9
  from database import engine, receipts
10
 
11
- # Load Hugging Face token from environment variables
12
  load_dotenv(override=True)
13
  hf_token = os.getenv("HF_TOKEN")
14
 
15
- # SQL Execution function
16
- @spaces.GPU
17
  @tool
18
  def sql_engine(query: str) -> str:
19
  """
@@ -38,13 +34,11 @@ def sql_engine(query: str) -> str:
38
  output = f"Error: {str(e)}"
39
  return output.strip()
40
 
41
- # Set up the Hugging Face agent
42
  agent = CodeAgent(
43
  tools=[sql_engine],
44
  model=HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", token=hf_token),
45
  )
46
 
47
- # Gradio function
48
  def query_sql(user_query):
49
  return sql_engine(user_query)
50
 
@@ -58,6 +52,5 @@ demo = gr.Interface(
58
  flagging_mode="never",
59
  )
60
 
61
- # Launch the app
62
  if __name__ == "__main__":
63
  demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
 
5
  from smolagents import tool, CodeAgent, HfApiModel
6
  import spaces
7
 
 
8
  from database import engine, receipts
9
 
 
10
  load_dotenv(override=True)
11
  hf_token = os.getenv("HF_TOKEN")
12
 
 
 
13
  @tool
14
  def sql_engine(query: str) -> str:
15
  """
 
34
  output = f"Error: {str(e)}"
35
  return output.strip()
36
 
 
37
  agent = CodeAgent(
38
  tools=[sql_engine],
39
  model=HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", token=hf_token),
40
  )
41
 
 
42
  def query_sql(user_query):
43
  return sql_engine(user_query)
44
 
 
52
  flagging_mode="never",
53
  )
54
 
 
55
  if __name__ == "__main__":
56
  demo.launch(server_name="0.0.0.0", server_port=7860, share=True)