Update README.md

README.md

@@ -30,20 +30,6 @@ This model is intended for the automatic generation of SQL queries from descript
 This model can be used with the Hugging Face Transformers library. Here is a quick example:
 
 ***************************
-
-from transformers import AutoTokenizer, AutoModelForCausalLM
-
-tokenizer = AutoTokenizer.from_pretrained("ansumanpandey/codgen-finetuned-SQLQueryGeneration")
-model = AutoModelForCausalLM.from_pretrained("ansumanpandey/codgen-finetuned-SQLQueryGeneration")
-
-def get_sql(query):
-  input_text = "Write a SQL query to %s </s>" % query
-  features = tokenizer([input_text], return_tensors='pt')
-
-  output = model.generate(input_ids=features['input_ids'],
-               attention_mask=features['attention_mask'],
-               max_new_tokens=70)
-  sql_query= tokenizer.decode(output[0])
-  return sql_query
-
+The link:
+https://huggingface.co/ansumanpandey/codgen-finetuned-SQLQueryGeneration/blob/main/The%20SQL%20Generation%20Code
 ***************************