biggan path

1 files changed, 2 insertions, 2 deletions

diff --git a/app/relay/modules/biggan.js b/app/relay/modules/biggan.js
index 6d6e211..44e8e4e 100644
--- a/app/relay/modules/biggan.js
+++ b/app/relay/modules/biggan.js
@@ -2,7 +2,7 @@ import path from 'path'
 import fs from 'fs'
 
 const name = 'biggan'
-const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/')
+const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/cli/')
 const env = {
   LD_LIBRARY_PATH: '/usr/local/cuda/lib64:' + process.env.TENSORRT_LIB_PATH,
   TFHUB_CACHE_DIR: '/home/lens/.tfhub',
@@ -11,7 +11,7 @@ const env = {
 }
 const live = {
   type: 'pytorch_biggan',
-  script: 'cli/cli.py',
+  script: 'cli.py',
   params: (task) => {
     console.log(task)
     const opt = task.opt || {}