Update app.py

fixing shape issue

app.py

@@ -61,12 +61,13 @@ def infer(prompts, negative_prompts, image):
         num_inference_steps=50,
         neg_prompt_ids=negative_prompt_ids,
         jit=True,
-    ).images[0]
+    ).images[0,0]
     
     #output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
     del image
     gc.collect()
     
+    output=np.array(output, dtype=np.float32)
     return output
 
 with gr.Blocks(theme='kfahn/AnimalPose') as demo: