From 890de23a0546f6b32e9573c12645122b36bfe6d2 Mon Sep 17 00:00:00 2001 From: Rishabh Srivastava Date: Wed, 7 Feb 2024 15:28:41 +0800 Subject: [PATCH] Update inference.py --- inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference.py b/inference.py index 75f8896..6539f05 100644 --- a/inference.py +++ b/inference.py @@ -58,7 +58,7 @@ def run_inference(question, prompt_file="prompt.md", metadata_file="metadata.sql if __name__ == "__main__": # Parse arguments - _default_question="Do we get more revenue from customers in New York compared to customers in San Francisco? Give me the total revenue for each city, and the difference between the two." + _default_question="Do we get more sales from customers in New York compared to customers in San Francisco? Give me the total sales for each city, and the difference between the two." parser = argparse.ArgumentParser(description="Run inference on a question") parser.add_argument("-q","--question", type=str, default=_default_question, help="Question to run inference on") args = parser.parse_args()