Update app.py

app.py

@@ -162,46 +162,26 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
             with gr.Column():
                 prompt = gr.Textbox(label="Prompt", value="portrait, color, cinematic")
                 id_image = gr.Image(label="ID Image")
-                id_weight = gr.Slider(0.0, 3.0, 1, step=0.05, label="id weight")
-
-                width = gr.Slider(256, 1536, 896, step=16, label="Width")
-                height = gr.Slider(256, 1536, 1152, step=16, label="Height")
-                num_steps = gr.Slider(1, 20, 20, step=1, label="Number of steps")
-                start_step = gr.Slider(0, 10, 0, step=1, label="timestep to start inserting ID")
-                guidance = gr.Slider(1.0, 10.0, 4, step=0.1, label="Guidance")
-                seed = gr.Textbox(-1, label="Seed (-1 for random)")
-                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", open=False):
-                    neg_prompt = gr.Textbox(
-                        label="Negative Prompt",
-                        value="bad quality, worst quality, text, signature, watermark, extra limbs")
-                    true_cfg = gr.Slider(1.0, 10.0, 1, step=0.1, label="true CFG scale")
-                    timestep_to_start_cfg = gr.Slider(0, 20, 1, step=1, label="timestep to start cfg", visible=args.dev)
-
                 generate_btn = gr.Button("Generate")
                 
             with gr.Column():
                 output_image = gr.Image(label="Generated Image")
-                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],
-                    ['white-haired woman with vr technology atmosphere, revolutionary exceptional magnum with remarkable details', 'example_inputs/liuyifei.png', 4, 4, 6349424134217931066, 1],
-                    ['a young child is eating Icecream', 'example_inputs/liuyifei.png', 4, 4, 10606046113565776207, 1],
-                    ['a man is holding a sign with text \"PuLID for FLUX\", winter, snowing, top of the mountain', 'example_inputs/pengwei.jpg', 4, 4, 2410129802683836089, 1],
-                    ['portrait, candle light', 'example_inputs/pengwei.jpg', 4, 4, 17522759474323955700, 1],
-                    ['profile shot dark photo of a 25-year-old male with smoke escaping from his mouth, the backlit smoke gives the image an ephemeral quality, natural face, natural eyebrows, natural skin texture, award winning photo, highly detailed face, atmospheric lighting, film grain, monochrome', 'example_inputs/pengwei.jpg', 4, 4, 17733156847328193625, 1],
-                    ['American Comics, 1boy', 'example_inputs/pengwei.jpg', 1, 4, 13223174453874179686, 1],
-                    ['portrait, pixar', 'example_inputs/pengwei.jpg', 1, 4, 9445036702517583939, 1],
-                    ['portrait, made of ice sculpture', 'example_inputs/lecun.jpg', 1, 1, 3811899118709451814, 5],
+                    ['a woman holding sign with glowing green text \"PuLID for FLUX\"', 'example_inputs/liuyifei.png'],
+                    ['portrait, side view', 'example_inputs/liuyifei.png'],
+                    ['white-haired woman with vr technology atmosphere', 'example_inputs/liuyifei.png'],
+                    ['a young child is eating Icecream', 'example_inputs/liuyifei.png'],
+                    ['a man is holding a sign with text \"PuLID for FLUX\", winter, snowing', 'example_inputs/pengwei.jpg'],
+                    ['portrait, candle light', 'example_inputs/pengwei.jpg'],
+                    ['profile shot dark photo of a 25-year-old male with smoke', 'example_inputs/pengwei.jpg'],
+                    ['American Comics, 1boy', 'example_inputs/pengwei.jpg'],
+                    ['portrait, pixar', 'example_inputs/pengwei.jpg'],
+                    ['portrait, made of ice sculpture', 'example_inputs/lecun.jpg'],
                 ]
 
                 example_images = [example[1] for example in all_examples]
@@ -219,19 +199,32 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                 )
 
                 def fill_example(evt: gr.SelectData):
-                    return [all_examples[evt.index][i] for i in [0, 1, 2, 3, 4, 5]]
+                    return [all_examples[evt.index][i] for i in [0, 1]]
 
                 gallery.select(
                     fill_example,
                     None,
-                    [prompt, id_image, start_step, guidance, seed, true_cfg],
+                    [prompt, id_image],
                 )
 
         generate_btn.click(
             fn=generate_image,
-            inputs=[width, height, num_steps, start_step, guidance, seed, prompt, id_image, id_weight, neg_prompt,
-                    true_cfg, timestep_to_start_cfg, max_sequence_length],
-            outputs=[output_image, seed_output, intermediate_output],
+            inputs=[
+                gr.Slider(256, 1536, 896, step=16, visible=False),  # width
+                gr.Slider(256, 1536, 1152, step=16, visible=False),  # height
+                gr.Slider(1, 20, 20, step=1, visible=False),  # num_steps
+                gr.Slider(0, 10, 0, step=1, visible=False),  # start_step
+                gr.Slider(1.0, 10.0, 4, step=0.1, visible=False),  # guidance
+                gr.Textbox(-1, visible=False),  # seed
+                prompt, 
+                id_image, 
+                gr.Slider(0.0, 3.0, 1, step=0.05, visible=False),  # id_weight
+                gr.Textbox("bad quality, worst quality, text, signature, watermark, extra limbs", visible=False),  # neg_prompt
+                gr.Slider(1.0, 10.0, 1, step=0.1, visible=False),  # true_cfg
+                gr.Slider(0, 20, 1, step=1, visible=False),  # timestep_to_start_cfg
+                gr.Slider(128, 512, 128, step=128, visible=False),  # max_sequence_length
+            ],
+            outputs=[output_image],
         )
 
     return demo