Spaces:
Runtime error
Runtime error
Caleb Fahlgren
commited on
Commit
·
4ed656e
1
Parent(s):
9c4b7af
add spaces gpu decorator
Browse files
app.py
CHANGED
|
@@ -8,6 +8,7 @@ import duckdb
|
|
| 8 |
import requests
|
| 9 |
import llama_cpp
|
| 10 |
import instructor
|
|
|
|
| 11 |
import enum
|
| 12 |
|
| 13 |
from pydantic import BaseModel, Field
|
|
@@ -82,6 +83,7 @@ CREATE TABLE {} (
|
|
| 82 |
return sql_ddl
|
| 83 |
|
| 84 |
|
|
|
|
| 85 |
def generate_query(dataset_id: str, query: str) -> str:
|
| 86 |
ddl = get_dataset_ddl(dataset_id)
|
| 87 |
|
|
|
|
| 8 |
import requests
|
| 9 |
import llama_cpp
|
| 10 |
import instructor
|
| 11 |
+
import spaces
|
| 12 |
import enum
|
| 13 |
|
| 14 |
from pydantic import BaseModel, Field
|
|
|
|
| 83 |
return sql_ddl
|
| 84 |
|
| 85 |
|
| 86 |
+
@spaces.GPU
|
| 87 |
def generate_query(dataset_id: str, query: str) -> str:
|
| 88 |
ddl = get_dataset_ddl(dataset_id)
|
| 89 |
|