Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

fix type

Browse Source
This commit is contained in:
Francisco Schulz 2023-02-15 16:46:47 +01:00
parent ee99d76aab
commit d4657f1ab1
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/serve.py
View File

@ -29,7 +29,7 @@ def process_request(request_message):
file_id = request_message["fileId"] file_id = request_message["fileId"]
target_file_name = f"{dossier_id}/{file_id}.{request_message['targetFileExtension']}" target_file_name = f"{dossier_id}/{file_id}.{request_message['targetFileExtension']}"
response_file_name = f"{dossier_id}/{file_id}.{request_message['responseFileExtension']}" response_file_name = f"{dossier_id}/{file_id}.{request_message['responseFileExtension']}"
logger.info("Processing file %s w/ file_id=%s, and daossier_id=%s", target_file_name, file_id, dossier_id) logger.info("Processing file %s w/ file_id=%s and dossier_id=%s", target_file_name, file_id, dossier_id)
bucket = PYINFRA_CONFIG.storage_bucket bucket = PYINFRA_CONFIG.storage_bucket
storage = get_storage(PYINFRA_CONFIG) storage = get_storage(PYINFRA_CONFIG)
@ -38,22 +38,22 @@ def process_request(request_message):
pipeline = load_pipeline(verbose=IMAGE_CONFIG.service.verbose, batch_size=IMAGE_CONFIG.service.batch_size) pipeline = load_pipeline(verbose=IMAGE_CONFIG.service.verbose, batch_size=IMAGE_CONFIG.service.batch_size)
if storage.exists(bucket, target_file_name): if storage.exists(bucket, target_file_name):
logger.info("fetching file for file_id=%s, and daossier_id=%s", file_id, dossier_id) logger.info("fetching file for file_id=%s and dossier_id=%s", file_id, dossier_id)
object_bytes = storage.get_object(bucket, target_file_name) object_bytes = storage.get_object(bucket, target_file_name)
object_bytes = gzip.decompress(object_bytes) object_bytes = gzip.decompress(object_bytes)
classifications = list(pipeline(pdf=object_bytes)) classifications = list(pipeline(pdf=object_bytes))
logger.info("predictions ready for file_id=%s, and daossier_id=%s", file_id, dossier_id) logger.info("predictions ready for file_id=%s and dossier_id=%s", file_id, dossier_id)
result = {**request_message, "data": classifications} result = {**request_message, "data": classifications}
storage_bytes = gzip.compress(json.dumps(result).encode("utf-8")) storage_bytes = gzip.compress(json.dumps(result).encode("utf-8"))
logger.info("storing predictions for file_id=%s, and daossier_id=%s", file_id, dossier_id) logger.info("storing predictions for file_id=%s and dossier_id=%s", file_id, dossier_id)
storage.put_object(bucket, response_file_name, storage_bytes) storage.put_object(bucket, response_file_name, storage_bytes)
return {"dossierId": dossier_id, "fileId": file_id} return {"dossierId": dossier_id, "fileId": file_id}
else: else:
logger.info("no files found for file_id=%s, and daossier_id=%s", file_id, dossier_id) logger.info("no files found for file_id=%s and dossier_id=%s", file_id, dossier_id)
return None return None