Chirayu commited on
Commit
fa5b868
1 Parent(s): f6f27da

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -6,7 +6,7 @@ tags:
6
  # What does this model do?
7
  This model converts the natural language input to Kusto (KQL) query. It is a fine-tuned CodeT5+ 220M. This model is a part of nl2query repository which is present at https://github.com/Chirayu-Tripathi/nl2query
8
 
9
- You can use this model via the github repository or via following code.
10
 
11
  ```python
12
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
@@ -15,6 +15,9 @@ model = AutoModelForSeq2SeqLM.from_pretrained("Chirayu/nl2kql")
15
  tokenizer = AutoTokenizer.from_pretrained("Chirayu/nl2kql")
16
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
  model = model.to(device)
 
 
 
18
  def generate_query(
19
  textual_query: str,
20
  num_beams: int = 10,
 
6
  # What does this model do?
7
  This model converts the natural language input to Kusto (KQL) query. It is a fine-tuned CodeT5+ 220M. This model is a part of nl2query repository which is present at https://github.com/Chirayu-Tripathi/nl2query
8
 
9
+ You can use this model via the github repository or via following code. More information can be found on the repository.
10
 
11
  ```python
12
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
15
  tokenizer = AutoTokenizer.from_pretrained("Chirayu/nl2kql")
16
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
  model = model.to(device)
18
+
19
+ textual_query = '''kusto: find the session ids which have duration greater than 10 and having Manoj Raheja as the owner | conferencesessions : conference, sessionid, session_title, session_type, owner, participants, URL, level, session_location, starttime, duration, time_and_duration, kusto_affinity'''
20
+
21
  def generate_query(
22
  textual_query: str,
23
  num_beams: int = 10,