Update app.py

app.py

@@ -30,17 +30,17 @@ def inference(source_prompt, target_prompt, source_guidance_scale=1, guidance_sc
     ratio = min(height / img.height, width / img.width)
     img = img.resize((int(img.width * ratio), int(img.height * ratio)))
 
-    result = pipe(prompt=target_prompt,
-                  source_prompt=source_prompt,
-                  init_image=img,
-                  num_inference_steps=num_inference_steps,
-                  eta=0.1,
-                  strength=strength,
-                  guidance_scale=guidance_scale,
-                  source_guidance_scale=source_guidance_scale,
-                  ).images[0]
+    results = pipe(prompt=target_prompt,
+                   source_prompt=source_prompt,
+                   init_image=img,
+                   num_inference_steps=num_inference_steps,
+                   eta=0.1,
+                   strength=strength,
+                   guidance_scale=guidance_scale,
+                   source_guidance_scale=source_guidance_scale,
+                   )
 
-    return replace_nsfw_images(result)
+    return replace_nsfw_images(results)
 
 
 def replace_nsfw_images(results):
@@ -60,7 +60,7 @@ with gr.Blocks(css=css) as demo:
                 <h1>CycleDiffusion with Stable Diffusion</h1>
               </div>
               <p>
-                Demo for CycleDiffusion with Stable Diffusion, built with Diffusers 🧨 by HuggingFace 🤗. <br>
+                Demo for CycleDiffusion with Stable Diffusion. <br>
                 <a href="https://huggingface.co/docs/diffusers/main/en/api/pipelines/cycle_diffusion">Pipeline doc</a> | <a href="https://arxiv.org/abs/2210.05559">Paper</a>
               </p>
               <p>You can skip the queue in the colab: <a href="https://colab.research.google.com/gist/ChenWu98/0aa4fe7be80f6b45d3d055df9f14353a/copy-of-fine-tuned-diffusion-gradio.ipynb"><img data-canonical-src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab" src="https://colab.research.google.com/assets/colab-badge.svg"></a></p>
@@ -103,6 +103,7 @@ with gr.Blocks(css=css) as demo:
 
                     with gr.Row():
                         seed = gr.Slider(0, 2147483647, label='Seed', value=0, step=1)
+                    with gr.Row():
                         generate = gr.Button(value="Edit")
 
     inputs = [source_prompt, target_prompt, source_guidance_scale, guidance_scale, num_inference_steps,
@@ -118,8 +119,9 @@ with gr.Blocks(css=css) as demo:
         image_out, inference, cache_examples=False)
 
     gr.Markdown('''
-      Space by: [![Twitter Follow](https://img.shields.io/twitter/follow/ChenHenryWu?style=social)](https://twitter.com/ChenHenryWu)
-        
+      Space built with Diffusers 🧨 by HuggingFace 🤗.
+      [<img src="https://img.shields.io/twitter/follow/ChenHenryWu?style=social">](https://twitter.com/ChenHenryWu)
+      
       ![visitors](https://visitor-badge.glitch.me/badge?page_id=ChenWu98.CycleDiffusion)
     ''')