upload bytesio object to cortex

1 files changed, 4 insertions, 1 deletions

diff --git a/inversion/image_inversion_placeholder.py b/inversion/image_inversion_placeholder.py
index 64929cc..88af8ef 100644
--- a/inversion/image_inversion_placeholder.py
+++ b/inversion/image_inversion_placeholder.py
@@ -9,8 +9,8 @@ import itertools
 import numpy as np
 import os
 import params
-import PIL
 import scipy
+import pickle
 import sys
 import tensorflow as tf
 import tensorflow_probability as tfp
@@ -485,6 +485,9 @@ for image_batch, label_batch in image_gen:
   out_labels[out_pos:out_pos+BATCH_SIZE] = label_batch
   out_err[out_pos:out_pos+BATCH_SIZE] = rec_err_batch
   out_pos += BATCH_SIZE
+  # upload image
+  # write encoding, latent to pkl file
+
   if params.max_batches > 0 and (out_pos / BATCH_SIZE) >= params.max_batches:
     break