LarkAI
/

codet5p-770m_nl2sql_oig

Text2Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

JacksonLark commited on May 23, 2023

Commit

cb5e623

•

1 Parent(s): caf9ac3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ model = T5ForConditionalGeneration.from_pretrained("LarkAI/codet5p-770m_nl2sql_o
 text = "Given the following schema:\ntrack (Track_ID, Name, Location, Seating, Year_Opened)\nrace (Race_ID, Name, Class, Date, Track_ID)\nWrite a SQL query to count the number of tracks."
 inputs = tokenizer.encode(text, return_tensors="pt").to(device)
 output_ids = model.generate(inputs, max_length=512)
-response_text = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
 # SELECT COUNT( * ) FROM track
 ```

 text = "Given the following schema:\ntrack (Track_ID, Name, Location, Seating, Year_Opened)\nrace (Race_ID, Name, Class, Date, Track_ID)\nWrite a SQL query to count the number of tracks."
 inputs = tokenizer.encode(text, return_tensors="pt").to(device)
 output_ids = model.generate(inputs, max_length=512)
+response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
 # SELECT COUNT( * ) FROM track
 ```