Update README.md

README.md

@@ -1,6 +1,7 @@
 ---
 library_name: transformers
-tags: []
+base_model:
+- Qwen/Qwen2.5-1.5B-Instruct
 ---
 
 # Model Card for Model ID
@@ -35,7 +36,47 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 
 ## Uses
 
-<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+context = """
+              
+"""
+
+input_question = ""
+
+prompt = f"""
+### Instruction:
+Based on the database schema description, generate the correct SQL query to answer the following question.
+
+### Context:
+{context}
+
+### Input:
+{input_question}
+
+### Output:
+"""
+
+tokens = tokenizer(prompt, return_tensors="pt").to("cuda")
+output = model.generate(
+    **tokens,
+    temperature=1,
+    max_new_tokens=100,
+    repetition_penalty=1.2,
+    top_p=0.9,
+    no_repeat_ngram_size=3
+)
+
+result = tokenizer.decode(output[0], skip_special_tokens=True)
+
+# Chỉ lấy phần sau "### Output:"
+import re
+
+output_start = re.search(r"### Output:\s*(.*)", result, re.DOTALL)
+if output_start:
+    result = output_start.group(1).strip()
+print(result)
+
+
+
 
 ### Direct Use