render video after live

1 files changed, 28 insertions, 2 deletions

diff --git a/app/relay/modules/biggan.js b/app/relay/modules/biggan.js
index 44e8e4e..74d308b 100644
--- a/app/relay/modules/biggan.js
+++ b/app/relay/modules/biggan.js
@@ -9,19 +9,45 @@ const env = {
   LC_ALL: 'C.UTF-8',
   LANG: 'C.UTF-8',
 }
+
+const invert = {
+  type: 'pytorch_biggan',
+  script: 'cli.py',
+  params: (task) => {
+    console.log(task)
+    return [
+      'biggan', 'extract_dense_vectors',
+      '--folder_id', task.opt.folder_id,
+      '--stochastic_clipping', '--label_clipping', '--clip_interval', 200
+    ]
+  }
+}
+
 const live = {
   type: 'pytorch_biggan',
   script: 'cli.py',
   params: (task) => {
     console.log(task)
     const opt = task.opt || {}
-    return ["biggan", "live"]
+    return ['biggan', 'live']
+  },
+  after: 'render',
+}
+
+const render = {
+  type: 'pytorch_biggan',
+  script: 'cli.py',
+  params: (task) => {
+    console.log(task)
+    return [
+      'cortex', 'upload_video',
+    ]
   }
 }
 
 export default {
   name, cwd, env,
   activities: {
-    live,
+    invert, live, render,
   }
 }