Upload 4 files

app.py

@@ -13,14 +13,15 @@ from huggingface_hub import snapshot_download
 tiny_model_dir = snapshot_download('wanderkid/unimernet_tiny')
 small_model_dir = snapshot_download('wanderkid/unimernet_small')
 base_model_dir = snapshot_download('wanderkid/unimernet_base')
-print("download done!")
-print(tiny_model_dir)
 
-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"))
+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"))
+# 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(), ".."))

cfg_base.yaml

@@ -2,13 +2,13 @@ model:
   arch: unimernet
   model_type: unimernet
   model_config:
-    model_name: ./models/unimernet_base
+    model_name: MODEL_DIR
     max_seq_len: 1536
 
   load_pretrained: True
-  pretrained: './models/unimernet_base/unimernet_base.pth'
+  pretrained: 'MODEL_DIR/unimernet_base.pth'
   tokenizer_config:
-    path: ./models/unimernet_base
+    path: MODEL_DIR
 
 datasets:
   formula_rec_eval:

cfg_small.yaml

@@ -2,13 +2,13 @@ model:
   arch: unimernet
   model_type: unimernet
   model_config:
-    model_name: ./models/unimernet_small
+    model_name: MODEL_DIR
     max_seq_len: 1536
 
   load_pretrained: True
-  pretrained: './models/unimernet_small/unimernet_small.pth'
+  pretrained: 'MODEL_DIR/unimernet_small.pth'
   tokenizer_config:
-    path: ./models/unimernet_small
+    path: MODEL_DIR
 
 datasets:
   formula_rec_eval:

cfg_tiny.yaml

@@ -2,13 +2,13 @@ model:
   arch: unimernet
   model_type: unimernet
   model_config:
-    model_name: ./models/unimernet_tiny
+    model_name: MODEL_DIR
     max_seq_len: 1536
 
   load_pretrained: True
-  pretrained: './models/unimernet_tiny/unimernet_tiny.pth'
+  pretrained: 'MODEL_DIR/unimernet_tiny.pth'
   tokenizer_config:
-    path: ./models/unimernet_tiny
+    path: MODEL_DIR
 
 datasets:
   formula_rec_eval: