Johannes commited on
Commit
b273a98
1 Parent(s): d3924a8

add 6B model back in

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. example_strings.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ description = """The NSQL model family was published by [Numbers Station](https:
37
  - [nsql-2B](https://huggingface.co/NumbersStation/nsql-2B)
38
  - [nsql-350M]((https://huggingface.co/NumbersStation/nsql-350M))
39
 
40
- This demo let's you choose from the 2B and 350M model and provides the three examples you can also find in their model cards.
41
 
42
  In general you should first provide the table schemas of the tables you have questions about and then prompt it with a natural language question.
43
  The model will then generate a SQL query that you can run against your database.
@@ -50,7 +50,7 @@ iface = gr.Interface(
50
  inputs=[gr.Text(label="Table schemas", placeholder="Insert your table schemas here"),
51
  gr.Text(label="Specify Task", value="Using valid SQLite, answer the following questions for the tables provided above."),
52
  gr.Text(label="Prompt", placeholder="Put your natural language prompt here"),
53
- gr.Dropdown(["nsql-2B", "nsql-350M"], value="nsql-6B")
54
  ],
55
  outputs="text",
56
  examples=[example1, example2, example3])
 
37
  - [nsql-2B](https://huggingface.co/NumbersStation/nsql-2B)
38
  - [nsql-350M]((https://huggingface.co/NumbersStation/nsql-350M))
39
 
40
+ This demo let's you choose from all of them and provides the three examples you can also find in their model cards.
41
 
42
  In general you should first provide the table schemas of the tables you have questions about and then prompt it with a natural language question.
43
  The model will then generate a SQL query that you can run against your database.
 
50
  inputs=[gr.Text(label="Table schemas", placeholder="Insert your table schemas here"),
51
  gr.Text(label="Specify Task", value="Using valid SQLite, answer the following questions for the tables provided above."),
52
  gr.Text(label="Prompt", placeholder="Put your natural language prompt here"),
53
+ gr.Dropdown(["nsql-6B", "nsql-2B", "nsql-350M"], value="nsql-6B")
54
  ],
55
  outputs="text",
56
  examples=[example1, example2, example3])
example_strings.py CHANGED
@@ -32,7 +32,7 @@ CREATE TABLE singer_in_concert (
32
  )""",
33
  "Using valid SQLite, answer the following questions for the tables provided above.",
34
  "What is the maximum, the average, and the minimum capacity of stadiums ?",
35
- "nsql-2B"]
36
 
37
  example2 = ["""CREATE TABLE stadium (
38
  stadium_id number,
 
32
  )""",
33
  "Using valid SQLite, answer the following questions for the tables provided above.",
34
  "What is the maximum, the average, and the minimum capacity of stadiums ?",
35
+ "nsql-6B"]
36
 
37
  example2 = ["""CREATE TABLE stadium (
38
  stadium_id number,