Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,14 @@ llm = Llama(
|
|
| 21 |
def generate_sql_query(text_input_schema: str, text_input_question: str):
|
| 22 |
try:
|
| 23 |
# Construct the prompt for the model
|
| 24 |
-
prompt = f"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
|
| 26 |
# Generate SQL query
|
| 27 |
completion = llm(
|
|
|
|
| 21 |
def generate_sql_query(text_input_schema: str, text_input_question: str):
|
| 22 |
try:
|
| 23 |
# Construct the prompt for the model
|
| 24 |
+
prompt = f"""
|
| 25 |
+
<|system|>
|
| 26 |
+
{text_input_schema}
|
| 27 |
+
|
| 28 |
+
<|user|>
|
| 29 |
+
{text_input_question}
|
| 30 |
+
|
| 31 |
+
<|sql|>"""
|
| 32 |
|
| 33 |
# Generate SQL query
|
| 34 |
completion = llm(
|