Update README.md

README.md

@@ -45,8 +45,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 model_name = "skshmjn/Llama-3.2-1B-Mongo-Instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
+schema = {} // Pass your mongodb schema here 
 
-prompt = "Find all employees older than 30 in the 'employees' collection."
+prompt = "Here is mongodb schema {schema} and Find all employees older than 30 in the 'employees' collection."
 inputs = tokenizer(prompt, return_tensors="pt")
 
 output = model.generate(**inputs, max_length=100)