Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import torch
+
 from ultralyticsplus import YOLO, render_result
 
 
@@ -40,18 +40,20 @@ def yoloV8_func(image: gr.Image = None,
     return render
 
 
+
 inputs = [
     gr.Image(type="filepath", label="Input Image"),
-    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, value=640,  # Changed 'default' to 'value'
+              step=32, label="Image Size"),
+    gr.Slider(minimum=0.0, maximum=1.0, value=0.25,  # Changed 'default' to 'value'
+              step=0.05, label="Confidence Threshold"),
+    gr.Slider(minimum=0.0, maximum=1.0, value=0.45,  # Changed 'default' to 'value'
+              step=0.05, label="IOU Threshold"),
 ]
 
 
-outputs = gr.outputs.Image( label="Output Image")
+
+output_image = gr.Image(label="Output Image")
 
 
 title = "YOLOv8 101: Custom Object Detection on Construction Workers"
@@ -64,7 +66,7 @@ examples = [['img1.jpg', 640, 0.5, 0.7],
 yolo_app = gr.Interface(
     fn=yoloV8_func,
     inputs=inputs,
-    outputs=outputs,
+    outputs=output_image,
     title=title,
     examples=examples,
     cache_examples=True,