Update README

app.py

@@ -36,12 +36,12 @@ def get_example():
         [
             get_image_path_list('./assets/aragaki_identity'),
             "An oil painting of a smiling woman.",
-            "A collage of images, monochrome, lowres, bad anatomy, worst quality, low quality",
+            "A collage of images, naked, monochrome, lowres, bad anatomy, worst quality, low quality",
         ],
         [
             get_image_path_list('./assets/blindbox_style'),
             "Donald Trump",
-            "A collage of images, monochrome, lowres, bad anatomy, worst quality, low quality",
+            "A collage of images, naked, monochrome, lowres, bad anatomy, worst quality, low quality",
         ],
     ]
     return case
@@ -72,10 +72,8 @@ easyref = EasyRef(pipe, multimodal_llm_path, ip_ckpt, device, num_tokens=64, use
 cv2.setNumThreads(1)
 
 @spaces.GPU(enable_queue=True)
-def generate_image(images, prompt, negative_prompt, height, width, scale, num_inference_steps, seed, progress=gr.Progress(track_tqdm=True)):
+def generate_image(images, prompt, negative_prompt, scale, num_inference_steps, seed, progress=gr.Progress(track_tqdm=True)):
     print("Generating")
-    new_width, new_height = adaptive_resize(width, height)
-    print("The output resolution is adaptively resized from {ori} to {new}".format(ori=res2string(width, height), new=res2string(new_width, new_height)))
     template = "Visualize a scene that closely resembles the provided images, capturing the essence and details described in this prompt:\n"
     system_prompt = [template + prompt, template]
     image = easyref.generate(
@@ -83,11 +81,10 @@ def generate_image(images, prompt, negative_prompt, height, width, scale, num_in
         system_prompt=system_prompt, 
         prompt=prompt, 
         negative_prompt=negative_prompt, 
+        scale=scale,
         num_samples=1,
         num_inference_steps=num_inference_steps,
-        seed=seed,
-        height=new_height,
-        width=new_width)    
+        seed=seed)
     print(image)
     return image
 
@@ -116,7 +113,7 @@ with gr.Blocks(css=css) as demo:
                         label="Upload several reference images",
                         file_types=["image"]
                     )
-            uploaded_files = gr.Gallery(label="Your images", visible=False, columns=5, rows=1, height=125)
+            uploaded_files = gr.Gallery(label="Your images", visible=False, columns=6, rows=1, height=125)
             with gr.Column(visible=False) as clear_button:
                 remove_and_reupload = gr.ClearButton(value="Remove and upload new ones", components=files, size="sm")
             prompt = gr.Textbox(label="Prompt",
@@ -125,28 +122,26 @@ with gr.Blocks(css=css) as demo:
             # style = gr.Radio(label="Generation type", info="For stylized try prompts like 'a watercolor painting of a woman'", choices=["Photorealistic", "Stylized"], value="Photorealistic")
             submit = gr.Button("Submit")
             with gr.Accordion(open=False, label="Advanced Options"):
-                height = gr.Slider(label="Height", info="This will be adaptively resized", value=1024, step=64, minimum=896, maximum=1280)
-                width = gr.Slider(label="Width", info="This will be adaptively resized", value=1024, step=64, minimum=896, maximum=1280)
-                scale = gr.Slider(label="Scale", info="Scale for reference images", value=1.0, step=0.1, minimum=0.5, maximum=1.5)
+                scale = gr.Slider(label="Scale", info="Scale for image reference", value=1.0, step=0.1, minimum=0.5, maximum=1.5)
                 num_inference_steps = gr.Slider(label="Number of inference steps", value=30, step=1, minimum=1, maximum=60)
                 seed = gr.Slider(label="Seed", value=24, step=1, minimum=0, maximum=MAX_SEED)
         with gr.Column():
             gallery = gr.Gallery(label="Generated Images")
+            gr.Examples(
+                examples=get_example(),
+                inputs=[files, prompt, negative_prompt],
+                run_on_click=True,
+                fn=upload_example_to_gallery,
+                outputs=[uploaded_files, clear_button, files],
+            )            
         # style.change(fn=change_style,
         #             inputs=style,
         #             outputs=[preserve, face_strength, likeness_strength])
         files.upload(fn=swap_to_gallery, inputs=files, outputs=[uploaded_files, clear_button, files])
         remove_and_reupload.click(fn=remove_back_to_files, outputs=[uploaded_files, clear_button, files])
         submit.click(fn=generate_image,
-                    inputs=[files,prompt,negative_prompt,height, width, scale, num_inference_steps, seed],
+                    inputs=[files, prompt, negative_prompt, scale, num_inference_steps, seed],
                     outputs=gallery)
-        gr.Examples(
-            examples=get_example(),
-            inputs=[files, prompt, negative_prompt],
-            run_on_click=True,
-            fn=upload_example_to_gallery,
-            outputs=[uploaded_files, clear_button, files],
-        )
             
     gr.Markdown("We release our checkpoints for research purposes only. Users are granted the freedom to create images using this tool, but they are expected to comply with local laws and utilize it in a responsible manner. The developers do not assume any responsibility for potential misuse by users.")