Update sql_generator.py

sql_generator.py

@@ -39,16 +39,16 @@ class SQLGenerator:
         ### Question: {natural_language_query}
         ### SQL Query:"""
         
-        inputs = self.tokenizer(prompt, return_tensors="pt", add_special_tokens=False).to(self.model.device)
         outputs = self.model.generate(
-            inputs["input_ids"],
-            max_length=256,
-            do_sample=False,
-            num_return_sequences=1,
-            eos_token_id=self.tokenizer.eos_token_id,
-            pad_token_id=self.tokenizer.pad_token_id
-        )
-        
+        inputs["input_ids"],
+        max_length=256,
+        do_sample=False,
+        num_return_sequences=1,
+        eos_token_id=self.tokenizer.eos_token_id,
+        pad_token_id=self.tokenizer.pad_token_id,
+        temperature=0.7,  # Adjust temperature for more creative output
+        top_k=50          # Consider top k predictions for variability
+    )
         return self.tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
     
     def fetch_shopify_data(self, endpoint):