Adding the pipeline_tag as Text Generation

README.md

@@ -5,57 +5,71 @@ license: llama2
 model_creator: NumbersStation
 model_name: NSQL Llama-2 7B
 model_type: llama
-prompt_template: '{prompt}
-
+prompt_template: |
+  {prompt}
 
   SELECT
-
-  '
 quantized_by: TheBloke
 widget:
 - example_title: Number stadiums
-  text: "CREATE TABLE stadium (\n    stadium_id number,\n    location text,\n    name\
-    \ text,\n    capacity number,\n)\n\n-- Using valid SQLite, answer the following\
-    \ questions for the tables provided above.\n\n-- how many stadiums in total?\n\
-    \nSELECT"
-- example_title: Open work orders
-  text: 'CREATE TABLE work_orders ( ID NUMBER, CREATED_AT TEXT, COST FLOAT, INVOICE_AMOUNT
-    FLOAT, IS_DUE BOOLEAN, IS_OPEN BOOLEAN, IS_OVERDUE BOOLEAN, COUNTRY_NAME TEXT,
+  text: >-
+    CREATE TABLE stadium (
+        stadium_id number,
+        location text,
+        name text,
+        capacity number,
     )
 
 
-    -- Using valid SQLite, answer the following questions for the tables provided
-    above.
+    -- Using valid SQLite, answer the following questions for the tables
+    provided above.
+
+
+    -- how many stadiums in total?
+
+
+    SELECT
+- example_title: Open work orders
+  text: >-
+    CREATE TABLE work_orders ( ID NUMBER, CREATED_AT TEXT, COST FLOAT,
+    INVOICE_AMOUNT FLOAT, IS_DUE BOOLEAN, IS_OPEN BOOLEAN, IS_OVERDUE BOOLEAN,
+    COUNTRY_NAME TEXT, )
+
+
+    -- Using valid SQLite, answer the following questions for the tables
+    provided above.
 
 
     -- how many work orders are open?
 
 
-    SELECT'
+    SELECT
 - example_title: Stadium capacity
-  text: 'CREATE TABLE stadium ( stadium_id number, location text, name text, capacity
+  text: >-
+    CREATE TABLE stadium ( stadium_id number, location text, name text, capacity
     number, highest number, lowest number, average number )
 
 
-    CREATE TABLE singer ( singer_id number, name text, country text, song_name text,
-    song_release_year text, age number, is_male others )
+    CREATE TABLE singer ( singer_id number, name text, country text, song_name
+    text, song_release_year text, age number, is_male others )
 
 
-    CREATE TABLE concert ( concert_id number, concert_name text, theme text, stadium_id
-    text, year text )
+    CREATE TABLE concert ( concert_id number, concert_name text, theme text,
+    stadium_id text, year text )
 
 
     CREATE TABLE singer_in_concert ( concert_id number, singer_id text )
 
 
-    -- Using valid SQLite, answer the following questions for the tables provided
-    above.
+    -- Using valid SQLite, answer the following questions for the tables
+    provided above.
 
 
     -- What is the maximum, the average, and the minimum capacity of stadiums ?
 
 
-    SELECT'
+    SELECT
+pipeline_tag: text-generation
 ---
 <!-- markdownlint-disable MD041 -->
 
@@ -475,4 +489,4 @@ print(tokenizer.decode(generated_ids[0], skip_special_tokens=True))
 
 For more information (e.g., run with your local database), please find examples in [this repository](https://github.com/NumbersStationAI/NSQL).
 
-<!-- original-model-card end -->
+<!-- original-model-card end -->