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 |
|