Update app.py

app.py

@@ -57,9 +57,9 @@ def infer(prompts, negative_prompts, image):
     output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
     return output_images
 
-e_images = ['examples/0.png',
-          'examples/1.png'
-          'examples/2.png']
+e_images = ['0.png',
+          '1.png'
+          '2.png']
 e_prompts = ['a dog in the middle of the road, shadow on the ground,light direction north-east',
            'a skyscraper in the middle of an intersection, shadow on the ground, light direction east',
            'a red rural house, light temperature 5500, shadow on the ground, light direction south-west']
@@ -76,7 +76,7 @@ with gr.Blocks() as demo:
       with gr.Column():
         in_image = gr.Image(label="Depth Map Conditioning")
       with gr.Column():
-        out_image = gr.Image(label="Generated Image")
+        out_image = gr.Gallery(label="Generated Image")
     with gr.Row():
         btn = gr.Button("Run")
         gr.Examples(examples=examples,