Commit
·
cda468b
1
Parent(s):
32c2894
update
Browse files- app-ngrok.py +5 -2
app-ngrok.py
CHANGED
@@ -4,6 +4,9 @@ import sqlparse
|
|
4 |
import requests
|
5 |
from time import sleep
|
6 |
import re
|
|
|
|
|
|
|
7 |
|
8 |
def format(text):
|
9 |
# Split the text by "|", and get the last element in the list which should be the final query
|
@@ -68,10 +71,10 @@ with gr.Blocks(theme='gradio/soft') as demo:
|
|
68 |
header = gr.HTML("""
|
69 |
<h1 style="text-align: center">SQL Skeleton WizardCoder Demo</h1>
|
70 |
<h3 style="text-align: center">🕷️☠️🧙♂️ Generate SQL queries from Natural Language 🕷️☠️🧙♂️</h3>
|
71 |
-
<p>Note: Should take around 30-60s to generate.</p>
|
72 |
""")
|
73 |
|
74 |
output_box = gr.Code(label="Generated SQL", lines=2, interactive=True)
|
|
|
75 |
input_text = gr.Textbox(lines=3, placeholder='Write your question here...', label='NL Input')
|
76 |
db_info = gr.Textbox(lines=4, placeholder='Example: | table_01 : column_01 , column_02 | table_02 : column_01 , column_02 | ...', label='Database Info')
|
77 |
|
@@ -91,7 +94,7 @@ with gr.Blocks(theme='gradio/soft') as demo:
|
|
91 |
["What are the number of concerts that occurred in the stadium with the largest capacity ?", "| stadium : stadium_id , location , name , capacity , highest , lowest , average | singer : singer_id , name , country , song_name , song_release_year , age , is_male | concert : concert_id , concert_name , theme , stadium_id , year | singer_in_concert : concert_id , singer_id | concert.stadium_id = stadium.stadium_id | singer_in_concert.singer_id = singer.singer_id | singer_in_concert.concert_id = concert.concert_id |"],
|
92 |
["How many male singers performed in concerts in the year 2023?", "| stadium : stadium_id , location , name , capacity , highest , lowest , average | singer : singer_id , name , country , song_name , song_release_year , age , is_male | concert : concert_id , concert_name , theme , stadium_id , year | singer_in_concert : concert_id , singer_id | concert.stadium_id = stadium.stadium_id | singer_in_concert.singer_id = singer.singer_id | singer_in_concert.concert_id = concert.concert_id |"],
|
93 |
["List the names of all singers who performed in a concert with the theme 'Rock'", "| stadium : stadium_id , location , name , capacity , highest , lowest , average | singer : singer_id , name , country , song_name , song_release_year , age , is_male | concert : concert_id , concert_name , theme , stadium_id , year | singer_in_concert : concert_id , singer_id | concert.stadium_id = stadium.stadium_id | singer_in_concert.singer_id = singer.singer_id | singer_in_concert.concert_id = concert.concert_id |"]
|
94 |
-
], inputs=[input_text, db_info, temperature, top_p, top_k, repetition_penalty], fn=generate, cache_examples=False if
|
95 |
|
96 |
quantized_model = "richardr1126/spider-skeleton-wizard-coder-ggml"
|
97 |
merged_model = "richardr1126/spider-skeleton-wizard-coder-merged"
|
|
|
4 |
import requests
|
5 |
from time import sleep
|
6 |
import re
|
7 |
+
import platform
|
8 |
+
|
9 |
+
print(f"Running on {platform.system()}")
|
10 |
|
11 |
def format(text):
|
12 |
# Split the text by "|", and get the last element in the list which should be the final query
|
|
|
71 |
header = gr.HTML("""
|
72 |
<h1 style="text-align: center">SQL Skeleton WizardCoder Demo</h1>
|
73 |
<h3 style="text-align: center">🕷️☠️🧙♂️ Generate SQL queries from Natural Language 🕷️☠️🧙♂️</h3>
|
|
|
74 |
""")
|
75 |
|
76 |
output_box = gr.Code(label="Generated SQL", lines=2, interactive=True)
|
77 |
+
note = gr.HTML("""<p style="font-size: 12px; text-align: center">⚠️ Should take 30-60s to generate</p>""")
|
78 |
input_text = gr.Textbox(lines=3, placeholder='Write your question here...', label='NL Input')
|
79 |
db_info = gr.Textbox(lines=4, placeholder='Example: | table_01 : column_01 , column_02 | table_02 : column_01 , column_02 | ...', label='Database Info')
|
80 |
|
|
|
94 |
["What are the number of concerts that occurred in the stadium with the largest capacity ?", "| stadium : stadium_id , location , name , capacity , highest , lowest , average | singer : singer_id , name , country , song_name , song_release_year , age , is_male | concert : concert_id , concert_name , theme , stadium_id , year | singer_in_concert : concert_id , singer_id | concert.stadium_id = stadium.stadium_id | singer_in_concert.singer_id = singer.singer_id | singer_in_concert.concert_id = concert.concert_id |"],
|
95 |
["How many male singers performed in concerts in the year 2023?", "| stadium : stadium_id , location , name , capacity , highest , lowest , average | singer : singer_id , name , country , song_name , song_release_year , age , is_male | concert : concert_id , concert_name , theme , stadium_id , year | singer_in_concert : concert_id , singer_id | concert.stadium_id = stadium.stadium_id | singer_in_concert.singer_id = singer.singer_id | singer_in_concert.concert_id = concert.concert_id |"],
|
96 |
["List the names of all singers who performed in a concert with the theme 'Rock'", "| stadium : stadium_id , location , name , capacity , highest , lowest , average | singer : singer_id , name , country , song_name , song_release_year , age , is_male | concert : concert_id , concert_name , theme , stadium_id , year | singer_in_concert : concert_id , singer_id | concert.stadium_id = stadium.stadium_id | singer_in_concert.singer_id = singer.singer_id | singer_in_concert.concert_id = concert.concert_id |"]
|
97 |
+
], inputs=[input_text, db_info, temperature, top_p, top_k, repetition_penalty], fn=generate, cache_examples=False if platform.system() == "Windows" or platform.system() == "Darwin" else True, outputs=output_box)
|
98 |
|
99 |
quantized_model = "richardr1126/spider-skeleton-wizard-coder-ggml"
|
100 |
merged_model = "richardr1126/spider-skeleton-wizard-coder-merged"
|