asoria HF staff commited on
Commit
d855f5c
·
verified ·
1 Parent(s): a970ae0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ headers = {
21
  }
22
  client = Client(headers=headers)
23
  api = HfApi(token=HF_TOKEN)
24
- pipe = pipeline("text-generation", model="motherduckdb/DuckDB-NSQL-7B-v0.1", device=1)
25
 
26
  @spaces.GPU
27
  def generate_sql(prompt):
 
21
  }
22
  client = Client(headers=headers)
23
  api = HfApi(token=HF_TOKEN)
24
+ pipe = pipeline("text-generation", model="motherduckdb/DuckDB-NSQL-7B-v0.1", device="cuda")
25
 
26
  @spaces.GPU
27
  def generate_sql(prompt):