Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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):
|