3 files changed, 5 insertions, 5 deletions
diff --git a/app/client/modules/biggan/biggan.tasks.js b/app/client/modules/biggan/biggan.tasks.js
index 2aed088..a28bf20 100644
--- a/app/client/modules/biggan/biggan.tasks.js
+++ b/app/client/modules/biggan/biggan.tasks.js
@@ -7,13 +7,11 @@ import actions from '../../actions'
 
 import module from './biggan.module'
 
-export const invert_task = (dataset, folder_id, epochs=1) => dispatch => {
-  dataset = dataset.name || dataset
+export const invert_task = (folder_id) => dispatch => {
   if (dataset === 'PLACEHOLDER') return
   const task = {
     module: module.name,
     activity: 'invert',
-    dataset,
     opt: {
       folder_id: folder_id,
     }
diff --git a/app/client/modules/biggan/views/biggan.show.js b/app/client/modules/biggan/views/biggan.show.js
index 8d85f87..22cecfb 100644
--- a/app/client/modules/biggan/views/biggan.show.js
+++ b/app/client/modules/biggan/views/biggan.show.js
@@ -35,7 +35,7 @@ class BigGANShow extends Component {
     }
   }
   render(){
-    const { biggan, match, history } = this.props
+    const { biggan, match, history, actions } = this.props
     const { folderLookup } = (biggan.data || {})
     const folder = (folderLookup || {})[biggan.folder_id] || {}
 
@@ -45,6 +45,7 @@ class BigGANShow extends Component {
           <div class='spaced'>
             <h1>{folder ? folder.name : <Loading />}</h1>
             <UploadStatus />
+            <button onClick={() => actions.tasks.invert_task(folder.id)}>Run inversion task</button>
           </div>
         </div>
         <div className='row'>
@@ -113,6 +114,7 @@ const mapStateToProps = state => ({
 
 const mapDispatchToProps = (dispatch, ownProps) => ({
   actions: bindActionCreators(bigganActions, dispatch),
+  tasks: bindActionCreators(bigganTasks, dispatch),
   remote: bindActionCreators(bigganTasks, dispatch),
 })
 
diff --git a/app/relay/modules/biggan.js b/app/relay/modules/biggan.js
index 9dfa6f7..ad26567 100644
--- a/app/relay/modules/biggan.js
+++ b/app/relay/modules/biggan.js
@@ -9,7 +9,7 @@ const env = {
 }
 const live = {
   type: 'pytorch_biggan',
-  script: 'inversion/live.py',
+  script: 'cli/cli.py biggan live',
   params: (task) => {
     console.log(task)
     const opt = task.opt || {}