Update app.py

app.py

@@ -89,22 +89,7 @@ u2net_model_path = os.path.join(CACHE_DIR, "u2netp.pth")
 yolo_world_global = None
 yolo_world_model_path = os.path.join(CACHE_DIR, "yolov8s_world.pt")  # Adjust path as needed
 
-def get_yolo_world():
-    """Lazy load YOLOWorld model"""
-    global yolo_world_global
-    if yolo_world_global is None:
-        logger.info("Loading YOLOWorld model...")
-        if os.path.exists(yolo_world_model_path):
-            try:
-                yolo_world_global = YOLOWorld(yolo_world_model_path)
-                logger.info("YOLOWorld model loaded successfully")
-            except Exception as e:
-                logger.error(f"Failed to load YOLOWorld: {e}")
-                yolo_world_global = None
-        else:
-            logger.warning("YOLOWorld model file not found, will raise error if used")
-            yolo_world_global = None
-    return yolo_world_global
+
     
 # Device configuration
 device = "cpu"
@@ -123,6 +108,11 @@ def ensure_model_files():
             shutil.copy("u2netp.pth", u2net_model_path)
         else:
             raise FileNotFoundError("u2netp.pth model file not found")
+    if not os.path.exists(yolo_world_model_path):
+        if os.path.exists("yolov8s-world.pt"):  # Adjust to match your file name
+            shutil.copy("yolov8s-world.pt", yolo_world_model_path)
+        else:
+            logger.warning("yolov8s-world.pt model file not found - falling back to full image processing")
 
 ensure_model_files()
 
@@ -144,7 +134,22 @@ def get_paper_detector():
             logger.warning("Paper model file not found, using fallback detection")
             paper_detector_global = None
     return paper_detector_global
-
+def get_yolo_world():
+    """Lazy load YOLOWorld model"""
+    global yolo_world_global
+    if yolo_world_global is None:
+        logger.info("Loading YOLOWorld model...")
+        if os.path.exists(yolo_world_model_path):
+            try:
+                yolo_world_global = YOLOWorld(yolo_world_model_path)
+                logger.info("YOLOWorld model loaded successfully")
+            except Exception as e:
+                logger.error(f"Failed to load YOLOWorld: {e}")
+                yolo_world_global = None
+        else:
+            logger.warning("YOLOWorld model file not found, will raise error if used")
+            yolo_world_global = None
+    return yolo_world_global
 def get_u2net():
     """Lazy load U2NETP model"""
     global u2net_global