refactor for new SDK version

supervisely-ecosystem · Jun 7, 2024 · dbad8f1 · dbad8f1
1 parent 10c2c9e
commit dbad8f1
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/supervisely_integration/serve/src/main.py b/supervisely_integration/serve/src/main.py
@@ -386,9 +386,9 @@ def smart_segmentation(response: Response, request: Request):
                     smtool_state,
                     api,
                     app_dir,
-                    cache_load_img=self.download_image,
-                    cache_load_frame=self.download_frame,
-                    cache_load_img_hash=self.download_image_by_hash,
+                    cache_load_img=self.cache.download_image,
+                    cache_load_frame=self.cache.download_frame,
+                    cache_load_img_hash=self.cache.download_image_by_hash,
                 )
                 self._inference_image_cache.set(hash_str, image_np)
             else:
@@ -469,7 +469,7 @@ def smart_segmentation(response: Response, request: Request):
         def is_online(response: Response, request: Request):
             response = {"is_online": True}
             return response
-        
+
         @server.post("/smart_segmentation_batched")
         def smart_segmentation_batched(response: Response, request: Request):
             response_batch = {}
@@ -486,7 +486,6 @@ def smart_segmentation_batched(response: Response, request: Request):
             return response_batch
 
 
-
 model = SegmentAnythingHQModel(
     use_gui=True,
     custom_inference_settings="./supervisely_integration/serve/custom_settings.yaml",