Spaces:
Runtime error
Runtime error
Mustehson
commited on
Commit
Β·
483db1c
1
Parent(s):
8b188e2
Change form app to Docker
Browse files
README.md
CHANGED
|
@@ -3,9 +3,8 @@ title: Datajoi Sql Agent
|
|
| 3 |
emoji: π£
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: purple
|
| 6 |
-
sdk:
|
| 7 |
sdk_version: 4.36.1
|
| 8 |
-
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
license: mit
|
| 11 |
---
|
|
|
|
| 3 |
emoji: π£
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: purple
|
| 6 |
+
sdk: docker
|
| 7 |
sdk_version: 4.36.1
|
|
|
|
| 8 |
pinned: false
|
| 9 |
license: mit
|
| 10 |
---
|
app.py
CHANGED
|
@@ -85,6 +85,7 @@ def get_prompt(schema, query_input):
|
|
| 85 |
# Generate SQL
|
| 86 |
@spaces.GPU
|
| 87 |
def generate_sql(prompt):
|
|
|
|
| 88 |
result = llama(prompt, temperature=0.1, max_tokens=1000)
|
| 89 |
return result["choices"][0]["text"]
|
| 90 |
|
|
|
|
| 85 |
# Generate SQL
|
| 86 |
@spaces.GPU
|
| 87 |
def generate_sql(prompt):
|
| 88 |
+
|
| 89 |
result = llama(prompt, temperature=0.1, max_tokens=1000)
|
| 90 |
return result["choices"][0]["text"]
|
| 91 |
|