Update app.py

app.py

@@ -34,8 +34,8 @@ api = HfApi(token=HF_TOKEN)
 
 
 # Second approach: Use transformers -> Took too much time
-pipe = pipeline("text-generation", model="motherduckdb/DuckDB-NSQL-7B-v0.1")
-pipe.to('cuda')
+pipe = pipeline("text-generation", model="motherduckdb/DuckDB-NSQL-7B-v0.1", device=1)
+# pipe.to('cuda')
 
 @spaces.GPU
 def query_local_model_transformers(prompt):