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
- internal_logg=app_logger,
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