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", dataDir: $.global.t2tWorkingDir + "/data",
lookupFile: $.global.t2tWorkingDir + "/code_search_index.csv", lookupFile: $.global.t2tWorkingDir + "/code_search_index.csv",
indexFile: $.global.t2tWorkingDir + "/code_search_index.nmslib", 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: [ args: [
"/usr/bin/tensorflow_model_server", "/usr/bin/tensorflow_model_server",
"--port=9000", "--port=9000",
"--rest_api_port=8000",
"--model_name=" + $.params.modelName, "--model_name=" + $.params.modelName,
"--model_base_path=" + $.params.modelPath, "--model_base_path=" + $.params.modelPath,
], ],
@ -342,7 +343,7 @@
gcpParts:: $.parts { gcpParts:: $.parts {
gcpEnv:: [ gcpEnv:: [
if $.gcpParams.gcpCredentialSecretName != "" then 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 { tfServingContainer: $.parts.tfServingContainer {