add big models back in

app.py

@@ -50,7 +50,7 @@ iface = gr.Interface(
     inputs=[gr.Text(label="Table schemas", placeholder="Insert your table schemas here"),
             gr.Text(label="Specify Task", value="Using valid SQLite, answer the following questions for the tables provided above."), 
             gr.Text(label="Prompt", placeholder="Put your natural language prompt here"), 
-            # gr.Dropdown(["nsql-6B", "nsql-2B", "nsql-350M"], value="nsql-6B")
+            gr.Dropdown(["nsql-6B", "nsql-2B", "nsql-350M"], value="nsql-6B")
             ], 
     outputs="text",
     examples=[example1, example2, example3])

example_strings.py

@@ -31,7 +31,8 @@ CREATE TABLE singer_in_concert (
     singer_id text
 )""",
 "Using valid SQLite, answer the following questions for the tables provided above.",
-"What is the maximum, the average, and the minimum capacity of stadiums ?"]
+"What is the maximum, the average, and the minimum capacity of stadiums ?",
+"nsql-350M"]
 
 example2 = ["""CREATE TABLE stadium (
     stadium_id number,
@@ -40,7 +41,8 @@ example2 = ["""CREATE TABLE stadium (
     capacity number,
 )""",
 "Using valid SQLite, answer the following questions for the tables provided above.",
-"how many stadiums in total?"]
+"how many stadiums in total?",
+"nsql-350M"]
 
 example3 = ["""CREATE TABLE work_orders (
     ID NUMBER,
@@ -53,4 +55,5 @@ example3 = ["""CREATE TABLE work_orders (
     COUNTRY_NAME TEXT,
 )""",
 "Using valid SQLite, answer the following questions for the tables provided above.",
-"how many work orders are open?"]
+"how many work orders are open?",
+"nsql-350M"]