Upload app.py

app.py

@@ -9,6 +9,18 @@ import numpy as np
 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')
+
+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"))
+shutil.move(small_model_dir, os.path.join(root_path, "models", "unimernet_small"))
+shutil.move(base_model_dir, os.path.join(root_path, "models", "unimernet_base"))
+# == download weights ==
+
 sys.path.insert(0, os.path.join(os.getcwd(), ".."))
 from unimernet.common.config import Config
 import unimernet.tasks as tasks
@@ -47,18 +59,6 @@ def gradio_reset():
     
 if __name__ == "__main__":
     root_path = os.path.abspath(os.getcwd())
-    
-    # == 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')
-    
-    os.makedirs(os.path.join(root_path, "models"), exist_ok=True)
-    shutil.move(tiny_model_dir, os.path.join(root_path, "models", "unimernet_tiny"))
-    shutil.move(small_model_dir, os.path.join(root_path, "models", "unimernet_small"))
-    shutil.move(base_model_dir, os.path.join(root_path, "models", "unimernet_base"))
-    # == download weights ==
-    
     # == load model ==
     model_tiny, vis_processor = load_model_and_processor(os.path.join(root_path, "cfg_tiny.yaml"))
     model_small, vis_processor = load_model_and_processor(os.path.join(root_path, "cfg_small.yaml"))