Update README.md

README.md

@@ -1,26 +1,26 @@
 # t5_wikisql_SQL2en
 
-This is the checkpoint for [t5_wikisql_SQL2en](https://huggingface.co/dbernsohn/t5_wikisql_SQL2en) after being fine-tunedon the wikisql dataset on t5-small.
+This is t5_wikisql_SQL2en , a t5-small fine-tuned version on the wikisql dataset.
 
 The model can be loaded like so:
 
 ```python
-from transformers import AutoModelWithLMHead, AutoTokenizer
-
-tokenizer = AutoTokenizer.from_pretrained("t5-small")
-model = AutoModelWithLMHead.from_pretrained("/content/model")
+from transformers import AutoTokenizer, AutoModelWithLMHead
+tokenizer = AutoTokenizer.from_pretrained("dbernsohn/t5_wikisql_SQL2en")
+model = AutoModelWithLMHead.from_pretrained("dbernsohn/t5_wikisql_SQL2en")
 ```
 
 You can then use this model to translate SQL queries into plain english.
 
 ```python
+query = "SELECT COUNT Params from model where location=HF-Hub"
 input_text = f"translate English to SQL: {query} </s>"
 features = tokenizer([input_text], return_tensors='pt')
 
-output = model.generate(input_ids=features['input_ids'], 
-                        attention_mask=features['attention_mask'])
+output = model.generate(input_ids=features['input_ids'].cuda(), 
+                        attention_mask=features['attention_mask'].cuda())
 
 tokenizer.decode(output[0])
 ```
 
-The whole training process and hyperparameters are in my []GitHub repo] (https://github.com/DorBernsohn)
+The whole training process and hyperparameters are in my [GitHub repo] (https://github.com/DorBernsohn)