Update app.py

app.py

@@ -108,11 +108,12 @@ def get_inpainted_img(img, mask):
 
 
 def remove_people(img, num_people_keep, dilate_kernel_size):
-
+    print('Obtaining mask...')
     mask = get_mask(img, num_people_keep, dilate_kernel_size)
-    print(mask.shape)
+    print('Mask obtained')
+    print('Inpainting with LAMA...')
     out = get_inpainted_img(img, mask)
-
+    print('Image Inpainted!')
     return out
 
 
@@ -131,6 +132,7 @@ model['lama'] = build_lama_model(lama_config, lama_ckpt, device=device)
      
 
 with gr.Blocks() as demo:
+    gr.Markdown(description)
     features = gr.State(None)
 
     with gr.Row():