File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
vision/automl/edge_container_predict Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change 37
37
38
38
# The absolute path of the current file. This will locate the model_path when
39
39
# run docker containers.
40
- ROOT_DIR = os .environ .get ('KOKORO_ROOT' , os .path .abspath (os .path .dirname (__file__ )))
40
+ ROOT_DIR = os .environ .get (
41
+ 'KOKORO_ROOT' , os .path .abspath (os .path .dirname (__file__ )))
41
42
MODEL_PATH = os .path .join (ROOT_DIR , 'model_path' )
42
43
43
44
IMAGE_FILE_PATH = os .path .join (os .path .dirname (__file__ ), 'test.jpg' )
44
45
# The cpu docker gcs path is from 'Edge container tutorial'.
45
- CPU_DOCKER_GCS_PATH = 'gcr.io/automl-vision-ondevice/gcloud-container-1.12.0:latest'
46
+ CPU_DOCKER_GCS_PATH = '{}' .format (
47
+ 'gcr.io/automl-vision-ondevice/gcloud-container-1.12.0:latest' )
46
48
# The path of a sample saved model.
47
- SAMPLE_SAVED_MODEL = 'gs://cloud-samples-data/vision/edge_container_predict/saved_model.pb'
49
+ SAMPLE_SAVED_MODEL = '{}' .format (
50
+ 'gs://cloud-samples-data/vision/edge_container_predict/saved_model.pb' )
48
51
# Container Name.
49
52
NAME = 'AutomlVisionEdgeContainerPredictTest'
50
53
# Port Number.
You can’t perform that action at this time.
0 commit comments