Update src/main.py

src/main.py

@@ -41,11 +41,11 @@ async def post_webhook(
     # ):
     # 	# no-op
     # 	return {"processed": False}
-    schedule_retrain(payload=payload)
-    # task_queue.add_task(
-    # 	schedule_retrain,
-    # 	payload
-    # )
+    # schedule_retrain(payload=payload)
+    task_queue.add_task(
+    	schedule_retrain,
+    	payload
+    )
 
     return {"processed": True}
 
@@ -62,7 +62,7 @@ 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])    # project = AutoTrain.create_project(payload)
     # AutoTrain.add_data(project_id=project["id"])
     # AutoTrain.start_processing(project_id=project["id"])
     except requests.HTTPError as err:
@@ -71,7 +71,7 @@ def schedule_retrain(payload: WebhookPayload):
         print(f"  {err.response.json()}")
         raise
     # Notify in the community tab
-    # notify_success(id)
+    notify_success(id)
     deploy_model(id="1726082187")
     print(result.returncode)
     return {"processed": True}