Add application file

app.py

@@ -18,13 +18,8 @@ def yoloV8_func(image: gr.Image = None,
                 image_size: gr.Slider = 640,
                 conf_threshold: gr.Slider = 0.4,
                 iou_threshold: gr.Slider = 0.50):
-    """This function performs YOLOv8 object detection on the given image.
-
-    Args:
-        image (gr.inputs.Image, optional): Input image to detect objects on. Defaults to None.
-        image_size (gr.inputs.Slider, optional): Desired image size for the model. Defaults to 640.
-        conf_threshold (gr.inputs.Slider, optional): Confidence threshold for object detection. Defaults to 0.4.
-        iou_threshold (gr.inputs.Slider, optional): Intersection over Union threshold for object detection. Defaults to 0.50.
+    """
+    This function performs YOLOv8 object detection on the given image.
     """
     # Load the YOLOv8 model from the 'best.pt' checkpoint
     model_path = "YOLO-best.pt"
@@ -49,11 +44,11 @@ def yoloV8_func(image: gr.Image = None,
 
 inputs = [
     gr.Image(type="filepath", label="Input Image"),
-    gr.Slider(minimum=320, maximum=1280, default=640,
+    gr.Slider(minimum=320, maximum=1280, value=640,
                      step=32, label="Image Size"),
-    gr.Slider(minimum=0.0, maximum=1.0, default=0.25,
+    gr.Slider(minimum=0.0, maximum=1.0, value=0.25,
                      step=0.05, label="Confidence Threshold"),
-    gr.Slider(minimum=0.0, maximum=1.0, default=0.45,
+    gr.Slider(minimum=0.0, maximum=1.0, value=0.45,
                      step=0.05, label="IOU Threshold"),
 ]