Update app.py

app.py

@@ -19,7 +19,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','linhcuem/chamdiem_yolov8_
              ['test_images/ee106392e56837366e79.jpg','linhcuem/chamdiem_yolov8_ver10', 640, 0.25, 0.45], ['test_images/f88d2214a4ee76b02fff.jpg','linhcuem/chamdiem_yolov8_ver10', 640, 0.25, 0.45]]
 
 # Load YOLO model
-# model = YOLO('linhcuem/chamdiem_yolov8_ver10')
+model = YOLO('linhcuem/chamdiem_yolov8_ver10')
 
 ###################################################
 def yolov8_img_inference(
@@ -32,7 +32,8 @@ def yolov8_img_inference(
     model = YOLO(model_path)
     model.conf = conf_threshold
     model.iou = iou_threshold
-    results = model.predict(image, imgsz=image_size, return_outputs=True)
+    #results = model.predict(image, imgsz=image_size, return_outputs=True)
+    results = model.predict(image)
     # object_prediction_list = []
     # for _, image_results in enumerate(results):
     #     if len(image_results)!=0: