Working search index server

This commit is contained in:
Sanyam Kapoor 2018-08-06 15:17:58 -07:00
parent 02db0065c1
commit 9ce23d9fc6
2 changed files with 3 additions and 2 deletions

View File

@ -97,7 +97,7 @@
dataDir: $.global.t2tWorkingDir + "/data",
lookupFile: $.global.t2tWorkingDir + "/code_search_index.csv",
indexFile: $.global.t2tWorkingDir + "/code_search_index.nmslib",
servingUrl: "http://localhost:8601",
servingUrl: "http://t2t-code-search.kubeflow:8000/v1/models/t2t-code-search:predict",
},
},
}

View File

@ -122,6 +122,7 @@
args: [
"/usr/bin/tensorflow_model_server",
"--port=9000",
"--rest_api_port=8000",
"--model_name=" + $.params.modelName,
"--model_base_path=" + $.params.modelPath,
],
@ -342,7 +343,7 @@
gcpParts:: $.parts {
gcpEnv:: [
if $.gcpParams.gcpCredentialSecretName != "" then
{ name: "GOOGLE_APPLICATION_CREDENTIALS", value: "/secret/gcp-credentials/key.json" },
{ name: "GOOGLE_APPLICATION_CREDENTIALS", value: "/secret/gcp-credentials/user-gcp-sa.json" },
],
tfServingContainer: $.parts.tfServingContainer {