Upload app.py

app.py

@@ -15,6 +15,9 @@ import spaces
 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("ls -l models/unimernet_tiny")
+os.system("ls -l models/unimernet_small")
+os.system("ls -l models/unimernet_base")
 # == download weights ==
 
 sys.path.insert(0, os.path.join(os.getcwd(), ".."))
@@ -122,10 +125,13 @@ def gradio_reset():
 if __name__ == "__main__":
     root_path = os.path.abspath(os.getcwd())
     # == load model ==
+    print("load tiny model ...")
     model_tiny, vis_processor = load_model_and_processor(os.path.join(root_path, "cfg_tiny.yaml"))
+    print("load small model ...")
     model_small, vis_processor = load_model_and_processor(os.path.join(root_path, "cfg_small.yaml"))
+    print("load base model ...")
     model_base, vis_processor = load_model_and_processor(os.path.join(root_path, "cfg_base.yaml"))
-    print("== load all models ==")
+    print("== load all models done. ==")
     # == load model ==
     
     with open("header.html", "r") as file: