bug: fix wrong usage of get_source_name() within infer_samgis_fn()
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ async def health() -> JSONResponse:
|
|
62 |
|
63 |
|
64 |
def infer_samgis_fn(request_input: ApiRequestBody | str) -> str | JSONResponse:
|
65 |
-
from samgis_web.web.web_helpers import
|
66 |
|
67 |
app_logger.info("starting inference request...")
|
68 |
try:
|
@@ -72,11 +72,10 @@ def infer_samgis_fn(request_input: ApiRequestBody | str) -> str | JSONResponse:
|
|
72 |
body_request = get_parsed_bbox_points_with_dictlist_prompt(request_input)
|
73 |
app_logger.info(f"body_request:{body_request}.")
|
74 |
try:
|
75 |
-
source_name =
|
76 |
-
app_logger.info(f"source_name = {source_name}.")
|
77 |
output = samexporter_predict(
|
78 |
bbox=body_request["bbox"], prompt=body_request["prompt"], zoom=body_request["zoom"],
|
79 |
-
source=body_request["source"], source_name=source_name, model_folder=model_folder
|
80 |
)
|
81 |
duration_run = time.time() - time_start_run
|
82 |
app_logger.info(f"duration_run:{duration_run}.")
|
|
|
62 |
|
63 |
|
64 |
def infer_samgis_fn(request_input: ApiRequestBody | str) -> str | JSONResponse:
|
65 |
+
from samgis_web.web.web_helpers import get_parsed_bbox_points_with_dictlist_prompt
|
66 |
|
67 |
app_logger.info("starting inference request...")
|
68 |
try:
|
|
|
72 |
body_request = get_parsed_bbox_points_with_dictlist_prompt(request_input)
|
73 |
app_logger.info(f"body_request:{body_request}.")
|
74 |
try:
|
75 |
+
app_logger.info(f"source_name = {body_request['source_name']}.")
|
|
|
76 |
output = samexporter_predict(
|
77 |
bbox=body_request["bbox"], prompt=body_request["prompt"], zoom=body_request["zoom"],
|
78 |
+
source=body_request["source"], source_name=body_request['source_name'], model_folder=model_folder
|
79 |
)
|
80 |
duration_run = time.time() - time_start_run
|
81 |
app_logger.info(f"duration_run:{duration_run}.")
|