Update app.py

app.py

@@ -181,7 +181,7 @@ def Aspergillus_Detect():
         "--source",
         test_path,
         "--weights",
-        "$WEIGHTS"
+        "$WEIGHTS",
         "--conf",
         "0.25",
         "--save-txt",
@@ -284,9 +284,9 @@ def classify_images(files):
     for file in all_files:
         print(file)
         
-    # if predict == 'Aspergillus':
-    #   result2 = Aspergillus_Detect()
-    #   return f'{result1}\n\n{result2}'
+    if predict == 'Aspergillus':
+      result2 = Aspergillus_Detect()
+      return f'{result1}\n\n{result2}'
     return result1
 
 import shutil