Update app.py

app.py

@@ -12,7 +12,7 @@ torch.hub.download_url_to_file('https://raw.githubusercontent.com/obss/sahi/main
 def yolov8_inference(
     image: gr.inputs.Image = None,
     model_path: gr.inputs.Dropdown = None,
-    image_size: gr.inputs.Slider = 640,
+    image_size: gr.inputs.Slider = 224,
     conf_threshold: gr.inputs.Slider = 0.25,
     iou_threshold: gr.inputs.Slider = 0.45,
 ):
@@ -63,7 +63,7 @@ inputs = [
     gr.inputs.Image(type="filepath", label="Input Image"),
     gr.inputs.Dropdown(["owaiskha9654/yolov8-custom_objects", "owaiskha9654/yolov8-custom_objects"], 
                        default="owaiskha9654/yolov8-custom_objects", label="Model"),
-    gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
+    gr.inputs.Slider(minimum=224, maximum=448, default=640, step=32, label="Image Size"),
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
 ]
@@ -71,7 +71,7 @@ inputs = [
 outputs = gr.outputs.Image(type="filepath", label="Output Image")
 title = "Ultralytics YOLOv8: State-of-the-Art YOLO Models"
 
-examples = [['highway.jpg', 'owaiskha9654/yolov8-custom_objects', 640, 0.25, 0.45], ['highway1.jpg', 'owaiskha9654/yolov8-custom_objects', 640, 0.25, 0.45],]# ['small-vehicles1.jpeg', 'owaiskha9654/yolov8-custom_objects', 1280, 0.25, 0.45]]
+examples = [['highway.jpg', 'owaiskha9654/yolov8-custom_objects', 224, 0.25, 0.45], ['highway1.jpg', 'owaiskha9654/yolov8-custom_objects', 224, 0.25, 0.45],]# ['small-vehicles1.jpeg', 'owaiskha9654/yolov8-custom_objects', 1280, 0.25, 0.45]]
 demo_app = gr.Interface(
     fn=yolov8_inference,
     inputs=inputs,