app

app.py

@@ -26,9 +26,9 @@ def run_evaluation(model_name):
 
     try:
         # Set up the arguments similar to the CLI in predict.py
-        dataset_path = "eval/data/dev.json"
-        table_meta_path = "eval/data/tables.json"
-        output_dir = "output/"
+        dataset_path = "duckdb-nsql/eval/data/dev.json"
+        table_meta_path = "duckdb-nsql/eval/data/tables.json"
+        output_dir = "duckdb-nsql/output/"
         prompt_format = "duckdbinstgraniteshort"
         stop_tokens = [';']
         max_tokens = 30000
@@ -88,7 +88,7 @@ def run_evaluation(model_name):
 
         # Set up evaluation arguments
         gold_path = Path(dataset_path)
-        db_dir = "eval/data/databases/"
+        db_dir = "duckdb-nsql/eval/data/databases/"
         tables_path = Path(table_meta_path)
 
         kmaps = test_suite_evaluation.build_foreign_key_map_from_json(str(tables_path))