2 files changed, 23 insertions, 0 deletions
diff --git a/app/relay/modules/biggan.js b/app/relay/modules/biggan.js
new file mode 100644
index 0000000..4a03117
--- /dev/null
+++ b/app/relay/modules/biggan.js
@@ -0,0 +1,21 @@
+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 live = {
+  type: 'pytorch',
+  script: 'live.py',
+  params: (task) => {
+    console.log(task)
+    const opt = task.opt || {}
+    return []
+  }
+}
+
+export default {
+  name, cwd, env,
+  activities: {
+    live,
+  }
+}
diff --git a/app/relay/modules/index.js b/app/relay/modules/index.js
index 2cbd08a..57b144e 100644
--- a/app/relay/modules/index.js
+++ b/app/relay/modules/index.js
@@ -1,3 +1,4 @@
+import biggan from './biggan'
 import morph from './morph'
 import pix2pix from './pix2pix'
 import pix2pixhd from './pix2pixhd'
@@ -7,6 +8,7 @@ import test from './test'
 // audio style transfer
 // cyclegan
 export default {
+  biggan,
   morph,
 	pix2pix,
   pix2wav: pix2pix,