Update app.py

app.py

@@ -94,7 +94,7 @@ def generate_image(
     negative="low quality",
     width=1024,
     height=1024,
-    scale=5,
+    scales=5,
     steps=30,
     strength=0.7,
     seed=-1,
@@ -121,23 +121,26 @@ def generate_image(
             prompt=text,
             image=init_image,
             negative_prompt=negative, 
-            guidance_scale=scale,
+            guidance_scale=scales,
             num_inference_steps=steps,
             strength=strength,
             generator = generator,
             num_images_per_prompt = nums,
-        )
+        ).images
     else:
         image = pipe(
             prompt=text,
             negative_prompt=negative, 
             width=width,
             height=height,
-            guidance_scale=scale,
+            guidance_scale=scales,
             num_inference_steps=steps,
             generator = generator,
             num_images_per_prompt = nums,
         ).images
+
+    print(image)
+    print(seed)
     return image, seed
 
 
@@ -159,8 +162,8 @@ examples = [
 # Gradio Interface
 
 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><b>Update</b>: fix T5 issue to support 512 token</center></p>")
+    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><b>Update</b>: fix diffuser to support 512 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=1)
@@ -184,7 +187,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                 value=1024,
             )
         with gr.Row():
-            scale = gr.Slider(
+            scales = gr.Slider(
                 label="Guidance",
                 minimum=3.5,
                 maximum=7,
@@ -212,7 +215,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                 maximum=10000000000000,
                 step=1,
                 value=-1,
-                scale=3,
+                scale=2,
             )
             nums = gr.Slider(
                 label="Image Numbers",
@@ -233,7 +236,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
 
     prompt.submit(fn=generate_image,
                  inputs=[prompt, negative, width, height, scale, steps, strength, seed, nums],
-                 outputs=[img,seed],
+                 outputs=[img, seed],
                  )