Skip to content

Commit

Permalink
Update detect.py (#2174)
Browse files Browse the repository at this point in the history
1) I got argument parse error when bucket_name=bucket_name is given
2)blob_list[0] gave me folder name
  • Loading branch information
agnelvishal authored and andrewferlitsch committed Jun 10, 2019
1 parent 2a1bf04 commit 0c0c1ed
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions vision/cloud-client/detect/detect.py
Original file line number Diff line number Diff line change
Expand Up @@ -704,7 +704,7 @@ def async_detect_document(gcs_source_uri, gcs_destination_uri):
bucket_name = match.group(1)
prefix = match.group(2)

bucket = storage_client.get_bucket(bucket_name=bucket_name)
bucket = storage_client.get_bucket(bucket_name)

# List objects with the given prefix.
blob_list = list(bucket.list_blobs(prefix=prefix))
Expand All @@ -715,7 +715,7 @@ def async_detect_document(gcs_source_uri, gcs_destination_uri):
# Process the first output file from GCS.
# Since we specified batch_size=2, the first response contains
# the first two pages of the input file.
output = blob_list[0]
output = blob_list[1]

json_string = output.download_as_string()
response = json_format.Parse(
Expand Down

0 comments on commit 0c0c1ed

Please sign in to comment.