Update app.py

app.py

@@ -74,7 +74,7 @@ def generate_image(
 
     if opts.seed is None:
         opts.seed = torch.Generator(device="cpu").seed()
-    print(f"Generating '{opts.prompt}' with seed {opts.seed}")
+
     t0 = time.perf_counter()
 
     use_true_cfg = abs(true_cfg - 1.0) > 1e-2
@@ -86,8 +86,6 @@ def generate_image(
         id_embeddings = None
         uncond_id_embeddings = None
 
-    print(id_embeddings)
-
     # prepare input
     x = get_noise(
         1,
@@ -97,7 +95,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,
@@ -142,10 +139,8 @@ def generate_image(
 
     t1 = time.perf_counter()
 
-    print(f"Done in {t1 - t0:.1f}s.")
     # bring into PIL format
     x = x.clamp(-1, 1)
-    # x = embed_watermark(x.float())
     x = rearrange(x[0], "c h w -> h w c")
 
     img = Image.fromarray((127.5 * (x + 1.0)).cpu().byte().numpy())
@@ -178,7 +173,7 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                 max_sequence_length = gr.Slider(128, 512, 128, step=128,
                                                 label="max_sequence_length for prompt (T5), small will be faster")
 
-                with gr.Accordion("Advanced Options (True CFG, true_cfg_scale=1 means use fake CFG, >1 means use true CFG, if using true CFG, we recommend set the guidance scale to 1)", open=False):    # noqa E501
+                with gr.Accordion("Advanced Options", open=False):
                     neg_prompt = gr.Textbox(
                         label="Negative Prompt",
                         value="bad quality, worst quality, text, signature, watermark, extra limbs")
@@ -192,12 +187,10 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                 seed_output = gr.Textbox(label="Used Seed")
                 intermediate_output = gr.Gallery(label='Output', elem_id="gallery", visible=args.dev)
 
-
         with gr.Row():
             with gr.Column():
                 gr.Markdown("## Examples")
                 
-                # 모든 예시를 하나의 리스트로 합칩니다
                 all_examples = [
                     ['a woman holding sign with glowing green text \"PuLID for FLUX\"', 'example_inputs/liuyifei.png', 4, 4, 2680261499100305976, 1],
                     ['portrait, side view', 'example_inputs/liuyifei.png', 4, 4, 1205240166692517553, 1],
@@ -211,11 +204,9 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                     ['portrait, made of ice sculpture', 'example_inputs/lecun.jpg', 1, 1, 3811899118709451814, 5],
                 ]
 
-                # 예시 이미지와 캡션을 준비합니다
                 example_images = [example[1] for example in all_examples]
                 example_captions = [example[0] for example in all_examples]
 
-                # Gallery 컴포넌트를 사용하여 그리드 형태로 예시들을 표시합니다
                 gallery = gr.Gallery(
                     value=list(zip(example_images, example_captions)),
                     label="Example Gallery",
@@ -227,7 +218,6 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                     height="auto"
                 )
 
-                # 예시 클릭 시 입력 필드에 값을 채우는 함수
                 def fill_example(evt: gr.SelectData):
                     return [all_examples[evt.index][i] for i in [0, 1, 2, 3, 4, 5]]