Update app.py

app.py

@@ -16,27 +16,27 @@ model = YOLO('best.pt')
 
 def predict_image(test_image):
     # save images
-    model.predict(test_image, save=True, conf=0.1, batch=2)
+    return model.predict(test_image, save=True, conf=0.1, batch=2, show=True)
 
-    path = "runs\\detect"
+    # 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]  #
+    # 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