Spaces:
Running
on
Zero
Running
on
Zero
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 |
|