feat: change output dir

Signed-off-by: 117503445 <[email protected]>
117503445 · Oct 14, 2023 · e2da53e · e2da53e
1 parent df90d9a
commit e2da53e
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/src/nougat/Dockerfile b/src/nougat/Dockerfile
@@ -7,6 +7,6 @@ WORKDIR /workspace/nougat-demo
 COPY . .
 
 # download models
-RUN nougat bitcoin.pdf -o out -m 0.1.0-base --markdown
+RUN nougat bitcoin-page-2.pdf -o out -m 0.1.0-base --markdown
 
 ENTRYPOINT [ "python", "fc.py" ]
diff --git a/src/nougat/bitcoin-page-2.pdf b/src/nougat/bitcoin-page-2.pdf
diff --git a/src/nougat/bitcoin.pdf b/src/nougat/bitcoin.pdf
diff --git a/src/nougat/fc.py b/src/nougat/fc.py
@@ -77,7 +77,7 @@ def create_main_logger():
     file.write_json(file_task, {"status": "executing"})
 
     try:
-        command = f'nougat {file} -o {dir_output} -m 0.1.0-base --markdown'
+        command = f'nougat {file} -o {dir_output / stem / "output"} -m 0.1.0-base --markdown'
         logger.debug(f'command = {command}')
         subprocess.run(command, shell=True)
         file.write_json(file_task, {"status": "done"})