Update README.md
Browse files
README.md
CHANGED
|
@@ -44,7 +44,7 @@ model_name = "NotShrirang/sql-deepseek-r1-distill-qwen-1.5B"
|
|
| 44 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 45 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 46 |
|
| 47 |
-
prompt = "
|
| 48 |
inputs = tokenizer(prompt, return_tensors="pt")
|
| 49 |
outputs = model.generate(**inputs, max_new_tokens=100)
|
| 50 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
|
|
|
| 44 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 45 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 46 |
|
| 47 |
+
prompt = "Write a SQL query to get the total revenue from the sales table."
|
| 48 |
inputs = tokenizer(prompt, return_tensors="pt")
|
| 49 |
outputs = model.generate(**inputs, max_new_tokens=100)
|
| 50 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|