Added examples to the interface

yolo-inference-project/src/app.py

@@ -20,6 +20,12 @@ def get_model_names():
 
 model_names = get_model_names()
 
+image_paths= [['examples/smaller_many_cans.jpg', 'best.pt', 0.5],
+              ['examples/gazebo_all.jpg', 'best.pt', 0.5],
+              ['examples/sequence_unity.jpg', 'best.pt', 0.5],
+              ['examples/real_plank.jpg', 'best.pt', 0.5],
+              ]
+
 demo = gr.Interface(
     fn=inference,
     inputs=[
@@ -29,7 +35,9 @@ demo = gr.Interface(
     ],
     outputs=gr.Image(type="numpy", label="Output Image"),
     title="YOLO Model Inference",
-    description="Select a YOLO model, upload an image, and set the confidence threshold to perform inference."
+    description="Select a YOLO model, upload an image, and set the confidence threshold to perform inference.",
+    examples=image_paths,
+    flagging_mode="auto"
 )
 
 demo.launch()

yolo-inference-project/src/model.py

@@ -1,7 +1,7 @@
 def load_model(model_path):
     from ultralytics import YOLO
     print(f"Loading model from {model_path}")
-    model = YOLO(model_path)  # Load the TorchScript model
+    model = YOLO(model_path)
     
     return model