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))
|