Added examples

app.py

@@ -1,3 +1,5 @@
+import os
+
 import gradio as gr
 from sample_solution import main as detect_bees
 
@@ -27,6 +29,15 @@ with gr.Blocks() as block:
 
     btn = gr.Button(value="Count the number of Bees")
     btn.click(detect_bees, inputs=[file_input], outputs=[file_output], queue=True)
+    examples_dir = os.listdir(os.path.join(os.path.dirname(__file__), "examples"))
+    print(examples_dir)
+    gr.Examples(
+        examples=[os.path.join(os.path.dirname(__file__), "examples", example_img) for example_img in examples_dir],
+        inputs=file_input,
+        outputs=file_output,
+        fn=detect_bees,
+        cache_examples=True,
+    )
 
 block.queue(concurrency_count=5).launch()
 # block.queue().launch()

sample_solution.py

@@ -260,12 +260,6 @@ def predict_bees(opt):
 
 
 def main(image):
-    # print(image)
-    # print(image.shape)
-    # print()
-    # cv2.imshow("sdfsd", image)
-    print("List of files: ", os.listdir("./"))
-    print("Model exists?: ", "custom_model.pt" in os.listdir("./"))
     image_path = os.path.join("images", str(time.time()) + ".jpg")
     image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
     cv2.imwrite(image_path, image)