Update app.py

app.py

@@ -16,27 +16,27 @@ model = YOLO('best.pt')
 
 def predict_image(test_image):
     # save images
-    return model.predict(test_image, save=True, conf=0.1, batch=2, show=True)
+    model.predict(test_image, save=True, conf=0.1, batch=2, show=True)
 
     path = "runs/detect"
     
-    # detection = (os.listdir(path))
+    detection = (os.listdir(path))
 
-    # def extract_digits(x):
-    #     digits = ''.join(filter(str.isdigit, x))
-    #     return int(digits) if digits else float('inf')  #Use float('inf') as a default value
+    def extract_digits(x):
+        digits = ''.join(filter(str.isdigit, x))
+        return int(digits) if digits else float('inf')  #Use float('inf') as a default value
 
-    # sorted_ = sorted(detection, key=extract_digits)
+    sorted_ = sorted(detection, key=extract_digits)
 
-    # if len(sorted_)>1:
-    #     latest_detection = sorted_[-2]  #
-    # else:
-    #     latest_detection = sorted_[0]
+    if len(sorted_)>1:
+        latest_detection = sorted_[-2]  # getting the latest predictions
+    else:
+        latest_detection = sorted_[0]
 
-    # detection_path = os.path.join(path, latest_detection)
+    detection_path = os.path.join(path, latest_detection)
 
-    # pred = cv2.imread(os.path.join(detection_path, test_image))
-    # return pred
+    pred = cv2.imread(os.path.join(detection_path, test_image))
+    return pred