Update app.py

app.py

@@ -86,7 +86,6 @@ def generate_image(
         id_embeddings = None
         uncond_id_embeddings = None
 
-    print(id_embeddings)
 
     # prepare input
     x = get_noise(
@@ -97,7 +96,6 @@ def generate_image(
         dtype=torch.bfloat16,
         seed=opts.seed,
     )
-    print(x)
     timesteps = get_schedule(
         opts.num_steps,
         x.shape[-1] * x.shape[-2] // 4,
@@ -203,7 +201,7 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                 generate_btn = gr.Button("Generate")
 
             with gr.Column():
-                output_image = gr.Image(label="Generated Image")
+                output_image = gr.Image(label="Generated Image", format='png')
                 seed_output = gr.Textbox(label="Used Seed")
                 intermediate_output = gr.Gallery(label='Output', elem_id="gallery", visible=args.dev)
                 gr.Markdown(_CITE_)