Update src/main.py

src/main.py

@@ -52,16 +52,17 @@ async def post_webhook(
 
 def schedule_retrain(payload: WebhookPayload):
     # Create the autotrain project
+    id = str(int(datetime.datetime.now().timestamp()))
     try:
         yaml_path = os.path.join(os.getcwd(), "src/config.yaml")
         with open(yaml_path) as f:
             list_doc = yaml.safe_load(f)
-            list_doc['project_name'] = str(int(datetime.datetime.now().timestamp()))
+            list_doc['project_name'] = id
 
         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:
@@ -70,7 +71,8 @@ def schedule_retrain(payload: WebhookPayload):
         print(f"  {err.response.json()}")
         raise
     # Notify in the community tab
-    notify_success('vicuna')
+    # notify_success(id)
+    deploy_model(id='1726082187')
     print(result.returncode)
     return {"processed": True}