Update app.py

app.py

@@ -90,7 +90,7 @@ def generate_image(
     
     if prompt['files']:
         #images = Image.open(prompt['files'][-1]).convert('RGB')
-        init_image = load_image(prompt['files'][-1]).resize((1024, 1024))
+        init_image = load_image(prompt['files'][-1]).resize((height, width))
     else:
         init_image = None
     generator = torch.Generator().manual_seed(seed)
@@ -114,6 +114,7 @@ def generate_image(
             guidance_scale=scale,
             num_inference_steps=steps,
             generator = generator,
+            max_length=512,
         )        
     return image.images[0]
 
@@ -137,10 +138,10 @@ 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/img-to-image generation<br>img2img keep 1024*1024</center></p>")
+    gr.HTML("<p><center><a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium'>sd3m</a> text/image-to-image generation</center></p>")
     with gr.Group():
         with gr.Row():
-            prompt = gr.MultimodalTextbox(label='Enter Your Prompt (Multi-Languages)', interactive=True, placeholder="Enter prompt,you can upload image.", file_types=['image'], scale=6)
+            prompt = gr.MultimodalTextbox(label='Enter Your Prompt (Multi-Languages)', interactive=True, placeholder="Enter prompt, add one image.", file_types=['image'], scale=6)
             submit = gr.Button(scale=1, variant='primary')
     img = gr.Image(label='SD3M Generated Image')
     with gr.Accordion("Advanced Options", open=False):