Update app.py

app.py

@@ -2,6 +2,7 @@ import numpy as np
 
 from dehazing_gen import CycleGenerator
 
+from PIL import Image
 import torch
 from torchvision import transforms
 
@@ -37,7 +38,9 @@ with gr.Blocks() as demo:
                 dehaze_button = gr.Button("Dehaze")
         with gr.Column():
             output_image = gr.Image(label="Output Image", type="pil")
-            for name, file in sample_images:
-                gr.Button(name).click(dehaze, inputs=input_image, outputs=output_image)
+            gr.Markdown("### Choose from these sample images below:")
+            for name, file_path in sample_images:
+                gr.Button(name).click(lambda fp=file_path:Image.open(fp), outputs=input_image)
+    dehaze_button.click(dehaze, inputs=input_image, outputs=output_image)
 
 demo.launch()