omeryentur commited on
Commit
62c8779
·
verified ·
1 Parent(s): f34b9ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
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"""<|system|> {text_input_schema} <|user|> {text_input_question} <|sql|>"""
 
 
 
 
 
 
 
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(