Update app.py

app.py

@@ -7,9 +7,9 @@ from ultralyticsplus import YOLO
 def yolov8_inference(
     image: gr.Image = None,
     model_path: gr.Dropdown = None,
-    image_size: gr.Slider = 640,
-    conf_threshold: gr.Slider = 0.25,
-    iou_threshold: gr.Slider = 0.45,
+    image_size: gr.Slider = gr.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
+    conf_threshold: gr.Slider = gr.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
+    iou_threshold: gr.Slider = gr.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
 ):
     """
     YOLOv8 inference function
@@ -23,7 +23,7 @@ def yolov8_inference(
     """
     model = YOLO(model_path)
     model.overrides['conf'] = conf_threshold
-    model.overrides['iou']= iou_threshold
+    model.overrides['iou'] = iou_threshold
     model.overrides['agnostic_nms'] = False  # NMS class-agnostic
     model.overrides['max_det'] = 1000 
     
@@ -47,10 +47,7 @@ inputs = [
 outputs = gr.Text(Class1)
 title = "AI-Powered Tire Quality Inspection: YOLOv8s Enhanced Classification"
 
-
-
-
-description="""
+description = """
 Welcome to our 🤖 AI-Powered Tire Quality Inspection Space – a cutting-edge solution harnessing the capabilities of YOLOv8s to revolutionize 🚗 tire quality control processes.
 """
 
@@ -65,4 +62,4 @@ demo_app = gr.Interface(
     cache_examples=True,
     theme='huggingface',
 )
-demo_app.queue().launch(debug=True)
+demo_app.queue().launch(debug=True)