Hugging Face's logo

Spaces:
asoria
/
datasets-text2sql
Sleeping

App Files Community
1
asoria HF staff commited on
Commit
bfe469b
·
verified ·
1 Parent(s): 86418eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,8 +34,8 @@ api = HfApi(token=HF_TOKEN)
34
 
35
 
36
  # Second approach: Use transformers -> Took too much time
37
- pipe = pipeline("text-generation", model="motherduckdb/DuckDB-NSQL-7B-v0.1")
38
- pipe.to('cuda')
39
 
40
  @spaces.GPU
41
  def query_local_model_transformers(prompt):
 
34
 
35
 
36
  # Second approach: Use transformers -> Took too much time
37
+ pipe = pipeline("text-generation", model="motherduckdb/DuckDB-NSQL-7B-v0.1", device=1)
38
+ # pipe.to('cuda')
39
 
40
  @spaces.GPU
41
  def query_local_model_transformers(prompt):