Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -109,8 +109,8 @@ def ensure_model_files():
|
|
109 |
else:
|
110 |
raise FileNotFoundError("u2netp.pth model file not found")
|
111 |
if not os.path.exists(yolo_world_model_path):
|
112 |
-
if os.path.exists("
|
113 |
-
shutil.copy("
|
114 |
else:
|
115 |
logger.warning("yolov8s-world.pt model file not found - falling back to full image processing")
|
116 |
|
|
|
109 |
else:
|
110 |
raise FileNotFoundError("u2netp.pth model file not found")
|
111 |
if not os.path.exists(yolo_world_model_path):
|
112 |
+
if os.path.exists("yolov8s_world.pt"): # Adjust to match your file name
|
113 |
+
shutil.copy("yolov8s_world.pt", yolo_world_model_path)
|
114 |
else:
|
115 |
logger.warning("yolov8s-world.pt model file not found - falling back to full image processing")
|
116 |
|