Update

app-ngrok.py

@@ -33,11 +33,13 @@ def format(text):
 
     return final_query_markdown
 
-def generate(input_message: str, db_info="", temperature=0.5, top_p=0.9, top_k=0, repetition_penalty=1.08, format_sql=True, stop_sequence="###,Explanation"):
+def generate(input_message: str, db_info="", temperature=0.3, top_p=0.9, top_k=0, repetition_penalty=1.08, format_sql=True, stop_sequence="###"):
     # Format the user's input message
     messages = f"Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\n\n### Instruction:\n\nConvert text to sql: {input_message} {db_info}\n\n### Response:\n\n"
 
     url = "https://e9f4be879d38-8269039109365193683.ngrok-free.app/api/v1/generate"
+    stop_sequence = stop_sequence.split(",")
+    stop = ["###"] + stop_sequence
     payload = {
         "prompt": messages,
         "temperature": temperature,
@@ -46,10 +48,10 @@ def generate(input_message: str, db_info="", temperature=0.5, top_p=0.9, top_k=0
         "top_a": 0,
         "n": 1,
         "max_context_length": 2048,
-        "max_length": 512,
+        "max_length": 256,
         "rep_pen": repetition_penalty,
         "sampler_order": [6,0,1,3,4,2,5],
-        "stop_sequence": stop_sequence.split(",").append("###"),
+        "stop_sequence": stop,
     }
     headers = {
         "Content-Type": "application/json",
@@ -76,7 +78,7 @@ def generate(input_message: str, db_info="", temperature=0.5, top_p=0.9, top_k=0
 
 # Gradio UI Code
 with gr.Blocks(theme='gradio/soft') as demo:
-    # Elements stack vertically by default just define variables in order you want them to stack
+    # Elements stack vertically by default just define elements in order you want them to stack
     header = gr.HTML("""
         <h1 style="text-align: center">SQL Skeleton WizardCoder Demo</h1>
         <h3 style="text-align: center">🕷️☠️🧙‍♂️ Generate SQL queries from Natural Language 🕷️☠️🧙‍♂️</h3>
@@ -88,14 +90,16 @@ with gr.Blocks(theme='gradio/soft') as demo:
     db_info = gr.Textbox(lines=4, placeholder='Example: | table_01 : column_01 , column_02 | table_02 : column_01 , column_02 | ...', label='Database Info')
 
     with gr.Accordion("Options", open=False):
-        temperature = gr.Slider(label="Temperature", minimum=0.0, maximum=1.0, value=0.5, step=0.1)
+        temperature = gr.Slider(label="Temperature", minimum=0.0, maximum=1.0, value=0.3, step=0.1)
         top_p = gr.Slider(label="Top-p (nucleus sampling)", minimum=0.0, maximum=1.0, value=0.9, step=0.01)
         top_k = gr.Slider(label="Top-k", minimum=0, maximum=200, value=0, step=1)
         repetition_penalty = gr.Slider(label="Repetition Penalty", minimum=1.0, maximum=2.0, value=1.08, step=0.01)
         format_sql = gr.Checkbox(label="Format SQL + Remove Skeleton", value=True, interactive=True)
-        stop_sequence = gr.Textbox(lines=1, value="Explanation,Results,Note", label='Extra Stop Sequence')
-        
+        stop_sequence = gr.Textbox(lines=1, value="Explanation,Note", label='Extra Stop Sequence')
+    
+    # Generate button UI element
     run_button = gr.Button("Generate SQL", variant="primary")
+    # When the button is clicked, call the generate function, inputs are taken from the UI elements, outputs are sent to outputs elements
     run_button.click(fn=generate, inputs=[input_text, db_info, temperature, top_p, top_k, repetition_penalty, format_sql, stop_sequence], outputs=output_box, api_name="txt2sql")
     
     info = gr.HTML(f"""