Update src/main.py

src/main.py

@@ -62,7 +62,9 @@ def schedule_retrain(payload: WebhookPayload):
         with open(yaml_path, "w") as f:
             yaml.dump(list_doc, f, default_flow_style=False)
 
-        result = Popen(['autotrain', '--config', yaml_path])    # project = AutoTrain.create_project(payload)
+        result = Popen(['autotrain', '--config', yaml_path])   
+        result.wait()
+        # project = AutoTrain.create_project(payload)
     # AutoTrain.add_data(project_id=project["id"])
     # AutoTrain.start_processing(project_id=project["id"])
     except requests.HTTPError as err: