base 64 conversion

generation.py

@@ -10,6 +10,8 @@ from shared import model_dict, sam_model_dict, DEFAULT_SO_NEGATIVE_PROMPT, DEFAU
 import gc
 from io import BytesIO
 import base64
+import pickle
+import codecs
 
 verbose = False
 
@@ -212,9 +214,12 @@ def run(
     gc.collect()
     torch.cuda.empty_cache()
 
-    with BytesIO() as buffer:
-        np.save(buffer, images[0])
-        img_str = base64.b64encode(buffer.getvalue()).decode('utf-8')
+    obj = np.random.normal(size=(10, 10))
+    obj_base64string = codecs.encode(pickle.dumps(obj, protocol=pickle.HIGHEST_PROTOCOL), "base64").decode('latin1')
+    img_str = pickle.loads(codecs.decode(obj_base64string.encode('latin1'), "base64"))
+    # with BytesIO() as buffer:
+    #     np.save(buffer, images[0])
+    #     img_str = base64.b64encode(buffer.getvalue()).decode('utf-8')
         
     return images[0], so_img_list, img_str