chore: update test prediction for QA

test_prediction.py

@@ -4,12 +4,14 @@ import utilities as us
 
 def main():
     model = ModelPrediction()
-    response = model.make_prediction(question='What is the name of Simone', db_schema='CREATE TABLE Player(Name, Age)', model_name="gpt-3.5", prompt='{question} {db_schema}')
+    response = model.make_prediction(
+        question='What is the surnamname of Simone',
+        db_schema='CREATE TABLE Player(Name, Age)\nINSERT INTO Player(Name, Age) VALUES (Simone, Papicchio)', 
+        model_name="gpt-3.5",
+        task='QA'
+    )
     print(response)  # dict[response, response_parsed, cost]
 
 
 if __name__ == "__main__":
-    # main()
-    us.check_and_create_dir('data/data_interface/')
-    us.check_and_create_dir('data/data_results/')
-    us.check_and_create_dir('data/databases/')
+    main()