From 8396f2ee6112d1692642d72d44d2eff831655529 Mon Sep 17 00:00:00 2001
From: Pavel Bartsits <48913536+cxnt@users.noreply.github.com>
Date: Thu, 9 May 2024 17:47:10 +0400
Subject: [PATCH] Update main.py

---
 serve/src/main.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/serve/src/main.py b/serve/src/main.py
index 067b1dd..949b810 100644
--- a/serve/src/main.py
+++ b/serve/src/main.py
@@ -35,9 +35,7 @@ class YOLOv8Model(sly.nn.inference.ObjectDetection):
 
     def initialize_custom_gui(self):
         """Create custom GUI layout for model selection. This method is called once when the application is started."""
-        self.pretrained_models_table = PretrainedModelsSelector(
-            yolov8_models, sort_tasks=False, sort_models=False
-        )
+        self.pretrained_models_table = PretrainedModelsSelector(yolov8_models)
         custom_models = sly.nn.checkpoints.yolov8.get_list(api, team_id)
         self.custom_models_table = CustomModelsSelector(
             team_id,