Spaces:
Sleeping
Sleeping
remove GPU decorator
Browse files
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)
|