From fe8ab6133a31284d94944f5d75da9d6ad2b2bdb9 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 18 Dec 2019 21:13:05 +0100
Subject: adding biggan

---
 app/relay/modules/biggan.js | 21 +++++++++++++++++++++
 app/relay/modules/index.js  |  2 ++
 2 files changed, 23 insertions(+)
 create mode 100644 app/relay/modules/biggan.js

(limited to 'app/relay/modules')

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,
-- 
cgit v1.2.3-70-g09d2