Update app.py

app.py

@@ -101,7 +101,7 @@ def generate(
         callback=callback_prior,
         callback_steps=callback_steps
     )
-
+    print(prior_output)
     if PREVIEW_IMAGES:
         for _ in range(len(DEFAULT_STAGE_C_TIMESTEPS)):
             r = next(prior_output)
@@ -119,7 +119,7 @@ def generate(
         generator=generator,
         output_type="pil",
     ).images
-
+    print(decoder_output)
     #Save images
     for image in decoder_output:
         user_history.save_image(
@@ -137,7 +137,7 @@ def generate(
                 "num_images_per_prompt": num_images_per_prompt,
             },
         )
-
+    
     yield decoder_output[0]