Update app.py

try just generated 1 image

app.py

@@ -58,10 +58,10 @@ def infer(prompts, negative_prompts, image):
         num_inference_steps=50,
         neg_prompt_ids=negative_prompt_ids,
         jit=True,
-    ).images
+    ).images[0]
     
-    output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
-    return output_images
+    #output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
+    #return output_images
 
 with gr.Blocks(theme='kfahn/AnimalPose') as demo:  
   gr.Markdown(
@@ -82,10 +82,13 @@ with gr.Blocks(theme='kfahn/AnimalPose') as demo:
     with gr.Column():
       #keypoint_tool = addp5sketch(sketch_url)
       keypoint_tool = gr.HTML(lines)
+      output = gr.Image(
+                label="Result",
+            )
           
-  run_btn.click(fn=infer, inputs = ["prompts", "negative_prompts", "conditioning_image"], outputs = "gallery")
+  run_btn.click(fn=infer, inputs = ["prompts", "negative_prompts", "conditioning_image"], outputs = output)
     
-#gr.Interface(fn=infer, inputs = ["text", "text", "image"], outputs = "gallery",
+#gr.Interface(fn=infer, inputs = ["text", "text", "image"], outputs = output,
             #examples=[["a Labrador crossing the road", "low quality", "myimage.jpg"]])   
     
 #with gr.Row():