aletrn commited on
Commit
368fc0b
·
1 Parent(s): e3fdb01

bug: fix wrong usage of get_source_name() within infer_samgis_fn()

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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 get_source_name, get_parsed_bbox_points_with_dictlist_prompt
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 = get_source_name(request_input.source_type)
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}.")