Update app.py

app.py

@@ -21,8 +21,8 @@ DEFAULT_NEGATIVE_PROMPT = (
 @spaces.GPU
 def run(*args):
     id_image = args[0]
-    supp_images = args[1:4]
-    prompt, neg_prompt, scale, n_samples, seed, steps, H, W, id_scale, mode, id_mix = args[4:]
+    supp_images = args[1]
+    prompt, neg_prompt, scale, n_samples, seed, steps, H, W, id_scale, mode, id_mix = args[2:]
 
     pipeline.debug_img_list = []
     if mode == 'fidelity':
@@ -59,7 +59,7 @@ def run(*args):
 
 
 _MARKDOWN_ = """
-This is the demo for ToonMagev2 for Image to Image Stylization
+<h2><b>This is the demo for ToonMagev2 for Image to Image Stylization</b></h2>
 
 **Tips**
 
@@ -72,6 +72,8 @@ This is the demo for ToonMagev2 for Image to Image Stylization
   If you find that the generated results are not stylized enough, you can choose the extremely style mode.
   
   Try out with different prompts using your image and do provide your feedback.
+
+Note: The image of the women is generated using AI, therefore it does not resemble any living person. 
   
 **Demo by [Sunder Ali Khowaja](https://sander-ali.github.io) - [X](https://x.com/SunderAKhowaja) -[Github](https://github.com/sander-ali) -[Hugging Face](https://huggingface.co/SunderAli17)**
 """
@@ -99,12 +101,12 @@ with gr.Blocks(title="ToonMagev2", js = js_func, theme = theme) as SAK:
                 supp_image1 = gr.Image(
                     label="Additional ID image (auxiliary)", sources="upload", type="numpy", height=256
                 )
-                supp_image2 = gr.Image(
-                    label="Additional ID image (auxiliary)", sources="upload", type="numpy", height=256
-                )
-                supp_image3 = gr.Image(
-                    label="Additional ID image (auxiliary)", sources="upload", type="numpy", height=256
-                )
+                # supp_image2 = gr.Image(
+                #     label="Additional ID image (auxiliary)", sources="upload", type="numpy", height=256
+                # )
+                # supp_image3 = gr.Image(
+                #     label="Additional ID image (auxiliary)", sources="upload", type="numpy", height=256
+                # )
             prompt = gr.Textbox(label="Prompt", value='portrait,cinematic,wolf ears,white hair')
             submit = gr.Button("Generate")
             neg_prompt = gr.Textbox(label="Negative Prompt", value=DEFAULT_NEGATIVE_PROMPT)
@@ -196,8 +198,8 @@ with gr.Blocks(title="ToonMagev2", js = js_func, theme = theme) as SAK:
     inps = [
         face_image,
         supp_image1,
-        supp_image2,
-        supp_image3,
+        # supp_image2,
+        # supp_image3,
         prompt,
         neg_prompt,
         scale,