mlbench123 commited on
Commit
375f3a6
·
verified ·
1 Parent(s): d054101

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("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
 
 
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