debug print threading ident

app.py

@@ -266,24 +266,31 @@ def make3d(images):
 
     return mesh_fpath, mesh_glb_fpath
 
-@rr.thread_local_stream("InstantMesh_visualization")
+@spaces.GPU
+def print_thread_ident_from_gpu():
+    print(threading.get_ident())
+
+@rr.thread_local_stream("InstantMesh")
 def log_to_rr(input_image, do_remove_background, sample_steps, sample_seed):
-    preprocessed_image = preprocess(input_image, do_remove_background)
 
-    stream = rr.binary_stream()
+    print(threading.get_ident())
+    print_thread_ident_from_gpu()
 
-    rr.log("preprocessed_image", rr.Image(preprocessed_image))
+    # preprocessed_image = preprocess(input_image, do_remove_background)
 
-    yield stream.read()
+    # stream = rr.binary_stream()
 
-    print(threading.get_ident())
-    z123_out = generate_mvs(input_image, sample_steps, sample_seed)
-    print(z123_out)
-    for image in z123_out.images:
-        rr.log("z123image", rr.Image(image))
-        yield stream.read()
-    yield stream.read()
-    pass
+    # rr.log("preprocessed_image", rr.Image(preprocessed_image))
+
+    # yield stream.read()
+
+    # z123_out = generate_mvs(input_image, sample_steps, sample_seed)
+    # print(z123_out)
+    # for image in z123_out.images:
+    #     rr.log("z123image", rr.Image(image))
+    #     yield stream.read()
+    # yield stream.read()
+    # pass
 
 _HEADER_ = '''
 <h2><b>Official 🤗 Gradio Demo</b></h2><h2><a href='https://github.com/TencentARC/InstantMesh' target='_blank'><b>InstantMesh: Efficient 3D Mesh Generation from a Single Image with Sparse-view Large Reconstruction Models</b></a></h2>