Spaces:

MLBench
/

Car-Mirror-Segmentation

Sleeping

Ayesha-Majeed commited on 12 days ago

Commit

876813a

verified ·

1 Parent(s): 01bd42f

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -522,7 +522,7 @@ def run_birefnet(img_rgb):
         model_path = None
         for p in paths_to_try:
-            if os.path.exists(p) and os.path.exists(os.path.join(p, "config.json")):
                 model_path = p
                 break

         model_path = None
         for p in paths_to_try:
+            if os.path.exists(p) and os.path.exists(os.path.join(p, "config.json")) and os.path.exists(os.path.join(p, "model.safetensors")):
                 model_path = p
                 break