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

rename publishing flag

Browse Source
This commit is contained in:
Julius Unverfehrt 2023-01-31 10:37:27 +01:00
parent 98dc001123
commit 4eb7f3c40a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/serve.py
View File

@ -37,9 +37,9 @@ def process_request(request_message):
pipeline = load_pipeline(verbose=IMAGE_CONFIG.service.verbose, batch_size=IMAGE_CONFIG.service.batch_size)
if not storage.exists(bucket, target_file_name):
should_publish_result = False
publish_result = False
else:
should_publish_result = True
publish_result = True
object_bytes = storage.get_object(bucket, target_file_name)
object_bytes = gzip.decompress(object_bytes)
classifications = list(pipeline(pdf=object_bytes))
@ -48,7 +48,7 @@ def process_request(request_message):
storage_bytes = gzip.compress(json.dumps(result).encode("utf-8"))
storage.put_object(bucket, response_file_name, storage_bytes)
return should_publish_result, {"dossierId": dossier_id, "fileId": file_id}
return publish_result, {"dossierId": dossier_id, "fileId": file_id}
def main():