Spaces:
Build error
Build error
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):
|