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_)