debug print

.gitignore

@@ -1,2 +1,3 @@
+__pycache__/
 *.nix
 venv/

app.py

@@ -11,6 +11,7 @@ from pytorch_lightning import seed_everything
 from omegaconf import OmegaConf
 from einops import rearrange, repeat
 from tqdm import tqdm
+import threading
 from typing import Any
 from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
 import rerun as rr
@@ -165,6 +166,8 @@ def pipeline_callback(pipe: Any, step_index: int, timestep: float, callback_kwar
 @spaces.GPU
 def generate_mvs(input_image, sample_steps, sample_seed):
 
+    print(threading.get_ident())
+
     seed_everything(sample_seed)
 
     return pipeline(
@@ -273,12 +276,13 @@ def log_to_rr(input_image, do_remove_background, sample_steps, sample_seed):
 
     yield stream.read()
 
+    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
 
 _HEADER_ = '''