1 files changed, 15 insertions, 0 deletions
diff --git a/app/client/modules/samplernn/samplernn.tasks.js b/app/client/modules/samplernn/samplernn.tasks.js
index bb22791..55828b8 100644
--- a/app/client/modules/samplernn/samplernn.tasks.js
+++ b/app/client/modules/samplernn/samplernn.tasks.js
@@ -22,6 +22,21 @@ export const train_task = (dataset, folder_id, epochs=1) => dispatch => {
   console.log(task)
   return actions.queue.add_task(task)
 }
+export const generate_task = (dataset, folder_id, sample_length=5, n_samples=6, epoch=0) => dispatch => {
+  const task = {
+    module: 'samplernn',
+    activity: 'generate',
+    dataset: dataset.name,
+    epoch: epoch || (dataset.checkpoints.length && dataset.checkpoints[0].epoch) || 0,
+    folder_id: folder_id,
+    opt: {
+      sample_length: 44100 * sample_length,
+      n_samples,
+    }
+  }
+  console.log(task)
+  return actions.queue.add_task(task)
+}
 export const fetch_task = (url, folder_id, file_id, dataset) => dispatch => {
   if (! url) return console.log('input file inaccessible (no url)')
   const task = {