1 files changed, 6 insertions, 1 deletions
diff --git a/app/relay/modules/samplernn.js b/app/relay/modules/samplernn.js
index bdbe602..2dacaa7 100644
--- a/app/relay/modules/samplernn.js
+++ b/app/relay/modules/samplernn.js
@@ -12,7 +12,12 @@ const fetch = {
   },
   listen: (task, line, i) => {
     // here i need to bridge again... get the filename that comes back from youtube-dl
-    // and tell the cortex that URL -> fn and add the filename!s
+    // and tell the cortex that URL -> fn and add the filename!
+    if ( line.match(/^got fn, /) ) {
+      let dataset = line.split(' => ')[1].trim()
+      task.dataset = dataset
+      return { type: 'progress', action: 'set_dataset', task, }
+    }
     return null
   }
 }