alessandro trinca tornidor
commited on
Commit
·
1e100ac
1
Parent(s):
e999e89
bug: fix wrong 'internal_logg' argument for app_helpers.get_inference_model_by_args()
Browse files
samgis_lisa_on_zero/prediction_api/lisa.py
CHANGED
@@ -10,7 +10,7 @@ from samgis_lisa_on_zero.utilities.constants import DEFAULT_URL_TILES, LISA_INFE
|
|
10 |
msg_write_tmp_on_disk = "found option to write images and geojson output..."
|
11 |
|
12 |
|
13 |
-
def load_model_and_inference_fn(inference_function_name_key):
|
14 |
from samgis_lisa_on_zero import app_logger
|
15 |
from lisa_on_cuda.utils import app_helpers
|
16 |
from samgis_lisa_on_zero.prediction_api.global_models import models_dict
|
@@ -20,7 +20,7 @@ def load_model_and_inference_fn(inference_function_name_key):
|
|
20 |
parsed_args = app_helpers.parse_args([])
|
21 |
inference_fn = app_helpers.get_inference_model_by_args(
|
22 |
parsed_args,
|
23 |
-
|
24 |
inference_decorator=SPACES_GPU
|
25 |
)
|
26 |
models_dict[inference_function_name_key]["inference"] = inference_fn
|
|
|
10 |
msg_write_tmp_on_disk = "found option to write images and geojson output..."
|
11 |
|
12 |
|
13 |
+
def load_model_and_inference_fn(inference_function_name_key: str):
|
14 |
from samgis_lisa_on_zero import app_logger
|
15 |
from lisa_on_cuda.utils import app_helpers
|
16 |
from samgis_lisa_on_zero.prediction_api.global_models import models_dict
|
|
|
20 |
parsed_args = app_helpers.parse_args([])
|
21 |
inference_fn = app_helpers.get_inference_model_by_args(
|
22 |
parsed_args,
|
23 |
+
internal_logger0=app_logger,
|
24 |
inference_decorator=SPACES_GPU
|
25 |
)
|
26 |
models_dict[inference_function_name_key]["inference"] = inference_fn
|