Update app.py

app.py

@@ -141,7 +141,7 @@ SEGMENTATION_MODELS = {
 }
 
 class YOLOWorldDetector:
-    def __init__(self, model_size="base"):
+    def __init__(self, model_size="small"):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.model_size = model_size
         self.model_name = DETECTION_MODELS[model_size]