Update app.py

app.py

@@ -30,6 +30,25 @@ label_path = Path(image_path,'labels')
 weight_path = Path(model_path,'best_Yolo_v7.pt')
 print(f'\nworking directory: {os.getcwd()}\n')
 
+
+import os
+
+# 指定工作目录
+working_directory = '/home/user/app'
+
+# 列出工作目录中的所有条目
+all_entries = os.listdir(working_directory)
+
+# 过滤出所有的目录
+all_directories = [entry for entry in all_entries if os.path.isdir(os.path.join(working_directory, entry))]
+
+# 打印所有的目录
+print("所有的目录：")
+for directory in all_directories:
+    print(directory)
+
+
+
 """# load MobileNet model"""
 
 MobileNet_model = keras.models.load_model(Path(model_path,'MobileNetV3Large_Genus_5th_fold_model'))