changed UI a bit

app.py

@@ -40,7 +40,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
 ip_model = ipown.IPAdapterFaceIDXL(pipe, ip_ckpt, device)
 
 @spaces.GPU(enable_queue=True)
-def generate_image(images, prompt, negative_prompt, preserve_face_structure, face_strength, likeness_strength, nfaa_negative_prompt, progress=gr.Progress(track_tqdm=True)):
+def generate_image(images, prompt, negative_prompt, preserve_face_structure, face_strength, likeness_strength, progress=gr.Progress(track_tqdm=True)):
     pipe.to(device)
     app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
     app.prepare(ctx_id=0, det_size=(640, 640))
@@ -52,29 +52,20 @@ def generate_image(images, prompt, negative_prompt, preserve_face_structure, fac
         faces = app.get(face)
         faceid_embed = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
         faceid_all_embeds.append(faceid_embed)
-        if(first_iteration and preserve_face_structure):
-            face_image = face_align.norm_crop(face, landmark=faces[0].kps, image_size=640) # you can also segment the face
-            first_iteration = False
-            
+
     average_embedding = torch.mean(torch.stack(faceid_all_embeds, dim=0), dim=0)
     
-    total_negative_prompt = f"{negative_prompt} {nfaa_negative_prompt}"
+    total_negative_prompt = negative_prompt
     
-    print("Generating normal")
+    print("Generating SDXL")
     image = ip_model.generate(
         prompt=prompt, negative_prompt=total_negative_prompt, faceid_embeds=average_embedding,
-        scale=likeness_strength, width=1024, height=1024, guidance_scale=7.5, num_inference_steps=30
+        scale=likeness_strength, width=1024, height=1024, guidance_scale=face_strength, num_inference_steps=30
     )
 
     print(image)
     return image
 
-def change_style(style):
-    if style == "Photorealistic":
-        return(gr.update(value=True), gr.update(value=1.3), gr.update(value=1.0))
-    else:
-        return(gr.update(value=True), gr.update(value=0.1), gr.update(value=0.8))
-
 def swap_to_gallery(images):
     return gr.update(value=images, visible=True), gr.update(visible=True), gr.update(visible=False)
 
@@ -85,7 +76,7 @@ h1{margin-bottom: 0 !important}
 '''
 with gr.Blocks(css=css) as demo:
     gr.Markdown("# IP-Adapter-FaceID SDXL demo")
-    gr.Markdown("Demo for the [h94/IP-Adapter-FaceID SDXL model](https://huggingface.co/h94/IP-Adapter-FaceID) - Non-commercial license")
+    gr.Markdown("My own Demo for the [h94/IP-Adapter-FaceID SDXL model](https://huggingface.co/h94/IP-Adapter-FaceID).")
     with gr.Row():
         with gr.Column():
             files = gr.Files(
@@ -99,18 +90,15 @@ with gr.Blocks(css=css) as demo:
                        info="Try something like 'a photo of a man/woman/person'",
                        placeholder="A photo of a [man/woman/person]...")
             negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="low quality")
-            style = gr.Radio(label="Generation type", info="For stylized try prompts like 'a watercolor painting of a woman'", choices=["Photorealistic", "Stylized"], value="Photorealistic")
+            style = "Photorealistic"
             submit = gr.Button("Submit")
-            with gr.Accordion(open=False, label="Advanced Options"):
-                preserve = gr.Checkbox(label="Preserve Face Structure", info="Higher quality, less versatility (the face structure of your first photo will be preserved). Unchecking this will use the v1 model.", value=True)
+            with gr.Accordion(open=True, label="Advanced Options"):
+                preserve = False
                 face_strength = gr.Slider(label="Face Structure strength", info="Only applied if preserve face structure is checked", value=1.3, step=0.1, minimum=0, maximum=3)
                 likeness_strength = gr.Slider(label="Face Embed strength", value=1.0, step=0.1, minimum=0, maximum=5)
                 nfaa_negative_prompts = gr.Textbox(label="Appended Negative Prompts", info="Negative prompts to steer generations towards safe for all audiences outputs", value="low quality, worst quality")    
         with gr.Column():
             gallery = gr.Gallery(label="Generated Images")
-        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,