mlbench123 commited on
Commit
458d3f0
·
verified ·
1 Parent(s): db79fa6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1079,16 +1079,16 @@ u2net_global = None
1079
  birefnet = None
1080
 
1081
  # Model paths - use absolute paths for Docker
1082
- reference_model_path = os.path.join(CACHE_DIR, "best.pt")
1083
  u2net_model_path = os.path.join(CACHE_DIR, "u2netp.pth")
1084
 
1085
  # Copy model files to cache if they don't exist - with error handling
1086
  def ensure_model_files():
1087
  if not os.path.exists(reference_model_path):
1088
- if os.path.exists("best.pt"):
1089
- shutil.copy("best.pt", reference_model_path)
1090
  else:
1091
- raise FileNotFoundError("best.pt model file not found")
1092
  if not os.path.exists(u2net_model_path):
1093
  if os.path.exists("u2netp.pth"):
1094
  shutil.copy("u2netp.pth", u2net_model_path)
 
1079
  birefnet = None
1080
 
1081
  # Model paths - use absolute paths for Docker
1082
+ reference_model_path = os.path.join(CACHE_DIR, "best1.pt")
1083
  u2net_model_path = os.path.join(CACHE_DIR, "u2netp.pth")
1084
 
1085
  # Copy model files to cache if they don't exist - with error handling
1086
  def ensure_model_files():
1087
  if not os.path.exists(reference_model_path):
1088
+ if os.path.exists("best1.pt"):
1089
+ shutil.copy("best1.pt", reference_model_path)
1090
  else:
1091
+ raise FileNotFoundError("best1.pt model file not found")
1092
  if not os.path.exists(u2net_model_path):
1093
  if os.path.exists("u2netp.pth"):
1094
  shutil.copy("u2netp.pth", u2net_model_path)