ansumanpandey commited on
Commit
2c5b451
1 Parent(s): 03a71d9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -11
README.md CHANGED
@@ -1,20 +1,15 @@
1
  ---
2
- language: en
3
  license: apache-2.0
4
- datasets:
5
- - custom_dataset_for_sql_generation
6
  tags:
7
- - natural-language-processing
8
  - code-generation
9
  - sql
10
  - transformers
11
  - finetuned-model
12
- model-index:
13
- - name: Finetuned CodeGen for SQL Query Generation
14
- results:
15
- - task:
16
- name: SQL Code Generation
17
- type: code-generation
18
 
19
 
20
  # Finetuned CodeGen for SQL Query Generation
@@ -47,4 +42,4 @@ def get_sql(query):
47
  attention_mask=features['attention_mask'],
48
  max_new_tokens=70)
49
  sql_query= tokenizer.decode(output[0])
50
- return sql_query
 
1
  ---
 
2
  license: apache-2.0
3
+ language:
4
+ - en
5
  tags:
6
+ - nlp
7
  - code-generation
8
  - sql
9
  - transformers
10
  - finetuned-model
11
+ ---
12
+
 
 
 
 
13
 
14
 
15
  # Finetuned CodeGen for SQL Query Generation
 
42
  attention_mask=features['attention_mask'],
43
  max_new_tokens=70)
44
  sql_query= tokenizer.decode(output[0])
45
+ return sql_query