update

infer_api.py

@@ -199,11 +199,15 @@ def inference(validation_pipeline, bkg_remover, input_image, vae, feature_extrac
                                   use_noise=use_noise, **validation,)
         out = rearrange(out, "B C f H W -> (B f) C H W", f=1)
 
+    print("OUT!!!!!!")
+
     img_buf = io.BytesIO()
     save_image(out[0], img_buf, format='PNG')
     img_buf.seek(0)
     img = Image.open(img_buf)
 
+    print("OUT2!!!!!!")
+
     torch.cuda.empty_cache()
     return img