Update app.py

app.py

@@ -78,6 +78,7 @@ def generate_image(
     height=1024,
     scale=5,
     steps=30,
+    strength=0.7,
     seed=-1):
 
     if seed == -1:
@@ -98,17 +99,18 @@ def generate_image(
 
     if init_image:
         image = pipe2(
-            text,
+            prompt_3=text,
             image=init_image,
-            negative_prompt=negative, 
+            negative_prompt_3=negative, 
             guidance_scale=scale,
             num_inference_steps=steps,
+            strength=strength,
             generator = generator,
         )
     else:
         image = pipe(
-            text,
-            negative_prompt=negative, 
+            prompt_3=text,
+            negative_prompt_3=negative, 
             width=width,
             height=height,
             guidance_scale=scale,
@@ -137,7 +139,7 @@ examples = [
 
 with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     gr.HTML("<h1><center>SD3M🦄</center></h1>")
-    gr.HTML("<p><center><a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium'>sd3m</a> text/image-to-image generation</center></p>")
+    gr.HTML("<p><center><a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium'>sd3m</a> text/image-to-image generation<br>Update: img2img, T5 long token</center></p>")
     with gr.Group():
         with gr.Row():
             prompt = gr.MultimodalTextbox(label='Enter Your Prompt (Multi-Languages)', interactive=True, placeholder="Enter prompt, add one image.", file_types=['image'], scale=6)
@@ -176,6 +178,13 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                 step=1,
                 value=30,
             )
+            strength = gr.Slider(
+                label="Strength",
+                minimum=0.0,
+                maximum=1.0,
+                step=0.1,
+                value=0.7,
+            )
         with gr.Row():
             seed = gr.Slider(
                 label="Seed (-1 Get Random)",
@@ -194,11 +203,11 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     )
 
     prompt.submit(fn=generate_image,
-                 inputs=[prompt, negative, width, height, scale, steps, seed],
+                 inputs=[prompt, negative, width, height, scale, steps, strength, seed],
                  outputs=img,
                  )
     submit.click(fn=generate_image,
-                 inputs=[prompt, negative, width, height, scale, steps, seed],
+                 inputs=[prompt, negative, width, height, scale, steps, strength, seed],
                  outputs=img,
                  )