Update app.py

app.py

@@ -49,8 +49,8 @@ if torch.cuda.is_available():
         previewer.load_state_dict(previewer_state_dict)
         def callback_prior(self_remote, i, t, kwargs):
             latents = kwargs["latents"]
-            print(latents)
             output = previewer(latents)
+            print(output)
             output = numpy_to_pil(output.clamp(0, 1).permute(0, 2, 3, 1).float().cpu().numpy())
             return output
         callback_steps = 1