diff --git a/src/serve.py b/src/serve.py index 4960563..d5cf053 100644 --- a/src/serve.py +++ b/src/serve.py @@ -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():