Upload 2 files

app.py

@@ -8,8 +8,7 @@ from numpy import random
 
 from models.experimental import attempt_load
 from utils.datasets import LoadImages
-from utils.general import (
-    check_img_size, non_max_suppression, scale_coords, xyxy2xywh)
+from utils.general import (non_max_suppression, scale_coords, xyxy2xywh)
 from utils.torch_utils import select_device
 
 import gradio as gr
@@ -109,10 +108,9 @@ if __name__ == '__main__':
             run_btn = gr.Button(variant="primary")
         
         with gr.Row():
-            examples_data = [["examples/Eda.png"],["examples/Fairies.png"]]
-            examples = gr.Dataset(components=[input_img], samples=examples_data)
-        
-        examples.click(lambda x: x[0], [examples], [input_img])
+            examples_data = [["examples/Eda.png"],["examples/Chtholly.png"],["examples/Fairies.png"]]
+            examples = gr.Examples(examples=examples_data,
+                                   inputs=input_img)
         run_btn.click(run, [input_img, crop_mode, tgt_width, tgt_height, face_ratio, threshold], [output_img])