isolate picke error

app.py

@@ -162,13 +162,13 @@ def pipeline_callback(output_queue: SimpleQueue, pipe: Any, step_index: int, tim
     image = pipe.vae.decode(latents / pipe.vae.config.scaling_factor, return_dict=False)[0]  # type: ignore[attr-defined]
     image = pipe.image_processor.postprocess(image, output_type="np").squeeze()  # type: ignore[attr-defined]
 
-    output_queue.put(("log", "mvs/image", rr.Image(image)))
-    output_queue.put(("log", "mvs/latents", rr.Tensor(latents.squeeze())))
+    # output_queue.put(("log", "mvs/image", rr.Image(image)))
+    # output_queue.put(("log", "mvs/latents", rr.Tensor(latents.squeeze())))
 
     return callback_kwargs
 
 @spaces.GPU
-def generate_mvs(input_image, sample_steps, sample_seed, output_queue: SimpleQueue):
+def generate_mvs(output_queue: SimpleQueue, input_image, sample_steps, sample_seed):
 
     seed_everything(sample_seed)
 
@@ -246,7 +246,7 @@ def make3d(output_queue: SimpleQueue, images: Image.Image):
                     render_size=render_size,
                 )['images_rgb']
 
-            print(f'type(framee)={type(frame)}')
+            print(f'type(frame)={type(frame)}')
             output_queue.put(("log", "3dvideo", rr.Image(frame)))
             # frames.append(frame)
 
@@ -272,13 +272,15 @@ def make3d(output_queue: SimpleQueue, images: Image.Image):
         vertices, faces, vertex_colors = mesh_out
         vertices = vertices[:, [1, 2, 0]]
         print(f'type(vertices)={type(vertices)}')
+        print(f'type(faces)={type(faces)}')
+        print(f'type(vertex_colors)={type(vertex_colors)}')
 
         save_glb(vertices, faces, vertex_colors, mesh_glb_fpath)
         save_obj(vertices, faces, vertex_colors, mesh_fpath)
         
         print(f"Mesh saved to {mesh_fpath}")
 
-    return mesh_fpath, mesh_glb_fpath
+    return mesh_out
 
 @rr.thread_local_stream("InstantMesh")
 def log_to_rr(input_image, do_remove_background, sample_steps, sample_seed):
@@ -292,34 +294,34 @@ def log_to_rr(input_image, do_remove_background, sample_steps, sample_seed):
 
     output_queue = SimpleQueue()
 
-    mvs_thread = threading.Thread(target=generate_mvs, args=[input_image, sample_steps, sample_seed, output_queue])
-    mvs_thread.start()
-
-    while True:
-        msg = output_queue.get()
-        if msg[0] == "z123_image":
-            z123_image = msg[1]
-            break
-        elif msg[0] == "log":
-            entity_path = msg[1]
-            entity = msg[2]
-            rr.log(entity_path, entity)
-            yield stream.read()
+    # mvs_thread = threading.Thread(target=generate_mvs, args=[output_queue, input_image, sample_steps, sample_seed])
+    # mvs_thread.start()
+
+    # while True:
+    #     msg = output_queue.get()
+    #     if msg[0] == "z123_image":
+    #         z123_image = msg[1]
+    #         break
+    #     elif msg[0] == "log":
+    #         entity_path = msg[1]
+    #         entity = msg[2]
+    #         rr.log(entity_path, entity)
+    #         yield stream.read()
     
-    mvs_thread.join()
+    # mvs_thread.join()
 
-    rr.log("z123image", rr.Image(z123_image))
-    yield stream.read()
+    # rr.log("z123image", rr.Image(z123_image))
+    # yield stream.read()
 
-    mesh_fpath, mesh_glb_fpath = make3d(output_queue, z123_image)
+    # mesh_fpath, mesh_glb_fpath = make3d(output_queue, z123_image)
 
-    while not output_queue.empty():
-        msg = output_queue.get()
-        if msg[0] == "log":
-            entity_path = msg[1]
-            entity = msg[2]
-            rr.log(entity_path, entity)
-            yield stream.read()
+    # while not output_queue.empty():
+    #     msg = output_queue.get()
+    #     if msg[0] == "log":
+    #         entity_path = msg[1]
+    #         entity = msg[2]
+    #         rr.log(entity_path, entity)
+    #         yield stream.read()
         
 
 _HEADER_ = '''