Spaces:
Running
Running
add GPU decorator
Browse files
app.py
CHANGED
|
@@ -3,6 +3,7 @@ from dotenv import load_dotenv
|
|
| 3 |
import gradio as gr
|
| 4 |
from sqlalchemy import text
|
| 5 |
from smolagents import tool, CodeAgent, HfApiModel
|
|
|
|
| 6 |
|
| 7 |
# Import the database setup from database.py
|
| 8 |
from database import engine, receipts
|
|
@@ -12,6 +13,7 @@ load_dotenv(override=True)
|
|
| 12 |
hf_token = os.getenv("HF_TOKEN")
|
| 13 |
|
| 14 |
# SQL Execution function
|
|
|
|
| 15 |
@tool
|
| 16 |
def sql_engine(query: str) -> str:
|
| 17 |
"""
|
|
|
|
| 3 |
import gradio as gr
|
| 4 |
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
|
|
|
|
| 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 |
"""
|