bug: fix wrong import for samgis inference function

samgis_lisa_on_zero/prediction_api/predictors.py

@@ -6,7 +6,7 @@ from samgis_lisa_on_zero.io.tms2geotiff import download_extent
 from samgis_lisa_on_zero.io.wrappers_helpers import check_source_type_is_terrain
 from samgis_lisa_on_zero.prediction_api.global_models import models_dict, embedding_dict
 from samgis_lisa_on_zero.utilities.constants import DEFAULT_URL_TILES, SLOPE_CELLSIZE
-from samgis_core.prediction_api.sam_onnx import SegmentAnythingONNX, get_raster_inference_with_embedding_from_dict
+from samgis_core.prediction_api import sam_onnx2, sam_onnx_inference
 from samgis_core.utilities.constants import MODEL_ENCODER_NAME, MODEL_DECODER_NAME, DEFAULT_INPUT_SHAPE
 from samgis_core.utilities.type_hints import LlistFloat, DictStrInt, ListDict
 
@@ -40,7 +40,7 @@ def samexporter_predict(
     """
     if models_dict[model_name_key]["instance"] is None:
         app_logger.info(f"missing instance model {model_name_key}, instantiating it now!")
-        model_instance = SegmentAnythingONNX(
+        model_instance = sam_onnx2.SegmentAnythingONNX2(
             encoder_model_path=MODEL_FOLDER / MODEL_ENCODER_NAME,
             decoder_model_path=MODEL_FOLDER / MODEL_DECODER_NAME
         )
@@ -63,7 +63,7 @@ def samexporter_predict(
         f"img type {type(img)} with shape/size:{img.size}, transform type: {type(transform)}, transform:{transform}.")
     app_logger.info(f"source_name:{source_name}, source_name type:{type(source_name)}.")
     embedding_key = f"{source_name}_z{zoom}_w{pt1[1]},s{pt1[0]},e{pt0[1]},n{pt0[0]}"
-    mask, n_predictions = get_raster_inference_with_embedding_from_dict(
+    mask, n_predictions = sam_onnx_inference.get_raster_inference_with_embedding_from_dict(
         img, prompt, models_instance, model_name_key, embedding_key, embedding_dict)
     app_logger.info(f"created {n_predictions} masks, preparing conversion to geojson...")
     return {