Update app.py

app.py

@@ -62,8 +62,7 @@ project_path = Path('./')
 model_path = Path(project_path,'Model')
 test_path = Path(project_path,'Test')
 os.makedirs(test_path, exist_ok=True)
-image_path = Path(project_path,'runs/detect/predict')
-label_path = Path(image_path,'labels')
+detect_path = Path(project_path,'runs/detect')
 weight_path = Path(model_path,'best_Yolo_v8n.pt')
 # 指定工作目录
 # working_directory = '/home/user/app'
@@ -162,14 +161,13 @@ def delete_files_in_folder(path):
             file_path = os.path.join(root, file)
             os.remove(file_path)
 
-# command = [
-#     "pip3",
-#     "install",
-#     "--no-cache-dir",
-#     "transformers"
-# ]
-# run_subprocess(command)
-
+def find_the_most_recent_predict_path(path):
+  path = list(map(lambda x: x.split('/'),path))
+  path = [i for i in path if 'predict' in i[-1]]
+  # Sort the list by the last element of each inner list
+  path = sorted(path, key=lambda x: x[-1])
+  return '/'.join(path[-1])
+    
 def Aspergillus_Detect():
     delete_files_in_folder(image_path)
     # get_all_files(working_directory)
@@ -177,7 +175,10 @@ def Aspergillus_Detect():
     model = YOLO(weight_path)  # pretrained YOLOv8n model
 
     model.predict(test_path, save = True , save_txt = True )  # predict on an image
-    get_all_files(working_directory)
+    all_files, all_directories = get_all_files(working_directory)
+
+    image_path = Path(find_the_most_recent_predict_path(all_directories))
+    label_path = Path(image_path,'labels')
     
     file_paths = []
     for root, dirs, files in os.walk(label_path):