Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 import sys
-# sys.path.append(os.path.abspath(os.path.dirname(os.getcwd())))
 # os.chdir("../")
 import gradio as gr
 import numpy as np
@@ -9,10 +8,8 @@ from matplotlib import pyplot as plt
 import torch
 import tempfile
 from lama_inpaint import inpaint_img_with_lama, build_lama_model, inpaint_img_with_builded_lama
-#from utils import load_img_to_array, save_array_to_img, dilate_mask, \
-#    show_mask, show_points
 from PIL import Image
-sys.path.insert(0, str(Path(__file__).resolve().parent / "third_party" / "segment-anything"))
+#sys.path.insert(0, str(Path(__file__).resolve().parent / "third_party" / "segment-anything"))
 import argparse
 
 import os
@@ -39,7 +36,7 @@ coco_metadata = MetadataCatalog.get("coco_2017_val")
 from detectron2_repo.projects.PointRend import point_rend
 
 
-title = "PeopleRemover"
+title = "# PeopleRemover"
 description = """
 In this space, you can remove the amount of people you want from a picture.
 ⚠️ This is just a demo version!
@@ -132,6 +129,7 @@ model['lama'] = build_lama_model(lama_config, lama_ckpt, device=device)
      
 
 with gr.Blocks() as demo:
+    gr.Markdown(title)
     gr.Markdown(description)
     features = gr.State(None)
 
@@ -164,6 +162,18 @@ with gr.Blocks() as demo:
         [img, features, img_out],
         [img, features, img_out]
     )
+        [img, num_people_keep, dilate_kernel_size],
+        [img_out]
+    gr.Examples(
+        examples=[os.path.join(os.path.dirname(__file__), "examples/002.jpg"),
+                  os.path.join(os.path.dirname(__file__), "examples/013.jpg"), 
+                  os.path.join(os.path.dirname(__file__), "examples/014.jpg"),
+                  os.path.join(os.path.dirname(__file__), "examples/015.jpg")]
+        inputs=img,
+        outputs=img_out,
+        fn=mirror,
+        cache_examples=True,
+    )
 
 if __name__ == "__main__":
     demo.launch()