ale
commited on
Commit
•
54a9099
1
Parent(s):
fe98e0b
[refactor] remove debug LogArgumentsDecorator
Browse files
src/prediction_api/predictors.py
CHANGED
@@ -10,12 +10,11 @@ from src.io.tms2geotiff import download_extent
|
|
10 |
from src.prediction_api.sam_onnx import SegmentAnythingONNX
|
11 |
from src.utilities.constants import MODEL_ENCODER_NAME, MODEL_DECODER_NAME, DEFAULT_TMS
|
12 |
from src.utilities.serialize import serialize
|
13 |
-
|
14 |
|
15 |
models_dict = {"fastsam": {"instance": None}}
|
16 |
|
17 |
|
18 |
-
@LogArgumentsDecorator.log_arguments_decorator
|
19 |
def samexporter_predict(bbox, prompt: list[dict], zoom: float, model_name: str = "fastsam") -> dict:
|
20 |
try:
|
21 |
if models_dict[model_name]["instance"] is None:
|
|
|
10 |
from src.prediction_api.sam_onnx import SegmentAnythingONNX
|
11 |
from src.utilities.constants import MODEL_ENCODER_NAME, MODEL_DECODER_NAME, DEFAULT_TMS
|
12 |
from src.utilities.serialize import serialize
|
13 |
+
|
14 |
|
15 |
models_dict = {"fastsam": {"instance": None}}
|
16 |
|
17 |
|
|
|
18 |
def samexporter_predict(bbox, prompt: list[dict], zoom: float, model_name: str = "fastsam") -> dict:
|
19 |
try:
|
20 |
if models_dict[model_name]["instance"] is None:
|