adding biggan

1 files changed, 21 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,
+  }
+}