Update app.py

app.py

@@ -100,11 +100,11 @@ def get_mask(img, num_people_keep, dilate_kernel_size):
 def get_inpainted_img(img, mask):
     lama_config = args.lama_config
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    out = []
+
     img_inpainted = inpaint_img_with_builded_lama(
         model['lama'], img, mask, lama_config, device=device)
-    out.append(img_inpainted)
-    return out
+
+    return img_inpainted
 
 
 def remove_people(img, num_people_keep, dilate_kernel_size):
@@ -137,7 +137,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column(scale=1):
-            img = gr.Image()# value="Input Image" .style(height="200px")
+            img = gr.Image(height=500)# value="Input Image" .style(height="200px")
             
             num_people_keep = gr.Number(label="Number of people to keep", minimum=0, maximum=100) 
             dilate_kernel_size = gr.Slider(label="Dilate Kernel Size", minimum=0, maximum=30, step=1, value=5)
@@ -146,7 +146,7 @@ with gr.Blocks() as demo:
             clear_button_image = gr.Button(value="Reset", variant="secondary", size="sm")#.style(full_width=True, size="sm")
             
         with gr.Column(scale=1):
-            img_out = gr.Image(interactive=False,show_download_button=True)# value="Image with People Removed", type="numpy", .style(height="200px")
+            img_out = gr.Image(height=500, interactive=False,show_download_button=True)# value="Image with People Removed", type="numpy", .style(height="200px")
     
     #mask = gr.outputs.Image(type="numpy", label="Segmentation Mask")#.style(height="200px")