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 = 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"),
+    image_size: int = 640,  # Set default value here
+    conf_threshold: float = 0.25,  # Set default value here
+    iou_threshold: float = 0.45,  # Set default value here
 ):
     """
     YOLOv8 inference function
@@ -39,9 +39,9 @@ inputs = [
     choices=["foduucom/Tyre-Quality-Classification-AI"], 
     value="foduucom/Tyre-Quality-Classification-AI", 
     label="Model"),
-    gr.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
-    gr.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
-    gr.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
+    gr.Slider(minimum=320, maximum=1280, step=32, label="Image Size"),
+    gr.Slider(minimum=0.0, maximum=1.0, step=0.05, label="Confidence Threshold"),
+    gr.Slider(minimum=0.0, maximum=1.0, step=0.05, label="IOU Threshold"),
 ]
 
 outputs = gr.Text(Class1)