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

add additional info logging

Browse Source
This commit is contained in:
Francisco Schulz 2023-02-14 15:01:25 +01:00
parent 6b75ef22c9
commit 4830b1cb91
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/serve.py
View File

@ -48,12 +48,14 @@ def process_request(request_message):
pipeline = load_pipeline(verbose=IMAGE_CONFIG.service.verbose, batch_size=IMAGE_CONFIG.service.batch_size)
if storage.exists(bucket, target_file_name):
logger.info("fetching file for file_id=%s, and daossier_id=%s", file_id, dossier_id)
# should_publish_result = True
object_bytes = storage.get_object(bucket, target_file_name)
object_bytes = gzip.decompress(object_bytes)
classifications = list(pipeline(pdf=object_bytes))
if storage.exists(bucket, figure_data_file_name):
logger.info("fetching figures for file_id=%s, and daossier_id=%s", file_id, dossier_id)
metadata_bytes = storage.get_object(bucket, figure_data_file_name)
metadata_bytes = gzip.decompress(metadata_bytes)
metadata_per_image = json.load(io.BytesIO(metadata_bytes))["data"]
@ -68,6 +70,7 @@ def process_request(request_message):
return {"dossierId": dossier_id, "fileId": file_id}
else:
logger.info("no files found for file_id=%s, and daossier_id=%s", file_id, dossier_id)
return None