Update app.py

app.py

@@ -10,13 +10,13 @@ from PIL import Image
 from huggingface_hub import snapshot_download
 
 # == download weights ==
-tiny_model_dir = snapshot_download('wanderkid/unimernet_tiny')
-small_model_dir = snapshot_download('wanderkid/unimernet_small')
-base_model_dir = snapshot_download('wanderkid/unimernet_base')
+tiny_model_dir = snapshot_download('wanderkid/unimernet_tiny', local_dir='./models/unimernet_tiny')
+small_model_dir = snapshot_download('wanderkid/unimernet_small', local_dir='./models/unimernet_small')
+base_model_dir = snapshot_download('wanderkid/unimernet_base', local_dir='./models/unimernet_base')
 
-os.system(f"sed -i 's/MODEL_DIR/{tiny_model_dir}/g' cfg_tiny.yaml")
-os.system(f"sed -i 's/MODEL_DIR/{small_model_dir}/g' cfg_small.yaml")
-os.system(f"sed -i 's/MODEL_DIR/{base_model_dir}/g' cfg_base.yaml")
+# os.system(f"sed -i 's/MODEL_DIR/{tiny_model_dir}/g' cfg_tiny.yaml")
+# os.system(f"sed -i 's/MODEL_DIR/{small_model_dir}/g' cfg_small.yaml")
+# os.system(f"sed -i 's/MODEL_DIR/{base_model_dir}/g' cfg_base.yaml")
 # root_path = os.path.abspath(os.getcwd())
 # os.makedirs(os.path.join(root_path, "models"), exist_ok=True)
 # shutil.move(tiny_model_dir, os.path.join(root_path, "models", "unimernet_tiny"))