Add musicsheet to E2E script

stanford-crfm · Mar 15, 2024 · 12b95bd · 12b95bd
1 parent 3108f28
commit 12b95bd
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/image2structure/end2end.py b/src/image2structure/end2end.py
@@ -98,6 +98,13 @@ def main():
                 f"{command_base} --tmp-path {args.destination_path}/tmp/{args.type}/{category} {latex_suffix}"
                 f" >> {args.destination_path}/logs/{args.type}/{category}.log"
             )
+    elif args.type == "musicsheet":
+        for i, category in enumerate(["music"]):
+            music_suffix: str = f"musicsheet --subcategory {category}"
+            collect_commands.append(
+                f"{command_base} --tmp-path {args.destination_path}/tmp/{args.type}/{category} {music_suffix}"
+                f" >> {args.destination_path}/logs/{args.type}/{category}.log"
+            )
     else:
         raise ValueError(f"Invalid type: {args.type}")
     os.makedirs(f"{args.destination_path}/logs/{args.type}", exist_ok=True)