Update train.py

src/server/train.py

@@ -13,6 +13,7 @@ import sqlite3
 DATA_SPLIT_FOLDER = "model_datasets"
 IMAGE_EXTENSION = ".png"  
 LOG_FILE = "training_logs.log"
+LATEST_DETECT = "latest_detect"
 LATEST_DETECT_IMAGES = "latest_detect/images"
 LATEST_DETECT_LABELS = "latest_detect/labels"
 image_folder=""
@@ -271,8 +272,8 @@ def train_yolo_model(
     call_reload_model_api()
     clear_images_source()   
 
-if(len(label_folder) >= 20):
-    copy_file_latest_training()
+if(len(label_files) >= 20):
+    if os.path.exists(LATEST_DETECT): copy_file_latest_training()
 
     # 🚀 Copy files
     train_copied_imgs, train_copied_lbls = move_files(train_files, train_img_folder, train_lbl_folder)