Update app.py

app.py

@@ -37,20 +37,16 @@ def get_all_files(directory):
     for root, dirs, files in os.walk(directory):
         for file in files:
             all_files.append(os.path.join(root, file))
+    # 打印所有文件路径
+    print("所有的文件：")
+    for file in all_files:
+        print(file)
     return all_files
 
 # 指定工作目录
 working_directory = '/home/user/app'
-
 # 列出工作目录及其子目录下的所有文件
-all_files = get_all_files(working_directory)
-
-# 打印所有文件路径
-print("所有的文件：")
-for file in all_files:
-    print(file)
-
-
+get_all_files(working_directory)
 
 """# load MobileNet model"""
 
@@ -271,7 +267,13 @@ def Aspergillus_Detect():
         a.append([i,round(df_pivot[i].sum()/detect_number,3)])
     a = sorted(a,key=lambda x: x[1],reverse=True)
 
-    print(f'a: {a}')
+    print(f'\na: {a}\n')
+
+    # 指定工作目录
+    working_directory = '/home/user/app'
+    # 列出工作目录及其子目录下的所有文件
+    get_all_files(working_directory)
+
     result = [f'Prediction of species of Aspergillus: {a[0][0] if a[0][1] >= threshold else "Unclassified"}',
       f'There are {df_pivot.shape[0]} mold images.',
             f'Yolov7 detects {detect_number} instances.',
@@ -289,10 +291,10 @@ def classify_images(files):
     # 列出工作目录及其子目录下的所有文件
     all_files = get_all_files(working_directory)
     
-    # 打印所有文件路径
-    print("所有的文件：")
-    for file in all_files:
-        print(file)
+    # # 打印所有文件路径
+    # print("所有的文件：")
+    # for file in all_files:
+    #     print(file)
         
     if predict == 'Aspergillus':
       result2 = Aspergillus_Detect()