Update app.py

app.py

@@ -9,7 +9,7 @@ from diffusers.utils import load_image
 from PIL import Image
 import requests
 import transformers
-from transformers import AutoTokenizer, T5EncoderModel
+from transformers import AutoConfig, AutoTokenizer, T5EncoderModel
 from translatepy import Translator
 
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
@@ -55,9 +55,15 @@ text_encoder_3 = T5EncoderModel.from_pretrained(
     torch_dtype=torch.float16,
 )
 
+model_config = AutoConfig.from_pretrained(
+    repo, 
+    subfolder="tokenizer_3",
+)
+
 tokenizer_3 = AutoTokenizer.from_pretrained(
     repo, 
-    subfolder="tokenizer_3", 
+    subfolder="tokenizer_3",
+    config=model_config,
     torch_dtype=torch.float16,
 )
 
@@ -100,9 +106,8 @@ def generate_image(
     if init_image:
         image = pipe2(
             prompt=text,
-            prompt_3=text,
             image=init_image,
-            negative_prompt_3=negative, 
+            negative_prompt=negative, 
             guidance_scale=scale,
             num_inference_steps=steps,
             strength=strength,
@@ -111,8 +116,7 @@ def generate_image(
     else:
         image = pipe(
             prompt=text,
-            prompt_3=text,
-            negative_prompt_3=negative, 
+            negative_prompt=negative, 
             width=width,
             height=height,
             guidance_scale=scale,
@@ -141,7 +145,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<br>Update: img2img, T5 long token</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, add Strength, 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)