Update README.md

README.md

@@ -77,7 +77,17 @@ model = AutoModelForCausalLM.from_pretrained("mudasir13cs/Field-adaptive-query-g
 tokenizer = AutoTokenizer.from_pretrained("mudasir13cs/Field-adaptive-query-generator")
 
 # Generate content
-input_text = "Generate 8 different search queries for: Title: Business Strategy Template, Description: Professional business strategy presentation, Industries: Business, Categories: Strategy, Tags: business, strategy, planning"
+# Format prompt using Gemma chat template
+input_text = """<start_of_turn>user
+Generate 8 different search queries that users might use to find this presentation template:
+    Title: Modern Business Presentation
+    Description: This modern business presentation template features a minimalist design...
+    Industries: Business, Marketing
+    Categories: Corporate, Professional
+    Tags: Modern, Clean, Professional
+<end_of_turn>
+<start_of_turn>model
+"""
 
 inputs = tokenizer(input_text, return_tensors="pt")
 outputs = model.generate(**inputs, max_length=512, temperature=0.7, do_sample=True)