Update app.py

app.py

@@ -324,7 +324,7 @@ class YOLOWorldDetector:
         return res_plotted, json_results
 
 # Initialize detector with default model
-detector = YOLOWorldDetector(model_size="base")
+detector = YOLOWorldDetector(model_size="small")
 
 def detection_inference(image, text_prompt, confidence, model_size):
     # Update model if needed
@@ -378,7 +378,7 @@ with gr.Blocks(title="YOLO Vision Suite", css=custom_css) as demo:
                             )
                             model_dropdown = gr.Dropdown(
                                 choices=list(DETECTION_MODELS.keys()),
-                                value="base",
+                                value="small",
                                 label="Model Size",
                                 elem_classes="gr-select"
                             )