Caleb Fahlgren commited on
Commit
4ed656e
1 Parent(s): 9c4b7af

add spaces gpu decorator

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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