Commit 21159040 authored by kernel.h@qq.com's avatar kernel.h@qq.com

修改找不到model文件时候的错误提示

parent 46c5d6e0
...@@ -23,6 +23,7 @@ python magicpdf.py --pdf /home/llm/Downloads/xxxx.pdf --model /home/llm/Downloa ...@@ -23,6 +23,7 @@ python magicpdf.py --pdf /home/llm/Downloads/xxxx.pdf --model /home/llm/Downloa
import os import os
import json as json_parse import json as json_parse
import sys
import click import click
from loguru import logger from loguru import logger
from pathlib import Path from pathlib import Path
...@@ -163,8 +164,8 @@ def pdf_command(pdf, model, method): ...@@ -163,8 +164,8 @@ def pdf_command(pdf, model, method):
if model is None: if model is None:
model = pdf.replace(".pdf", ".json") model = pdf.replace(".pdf", ".json")
if not os.path.exists(model): if not os.path.exists(model):
print(f"make sure json file existed and place under {os.dirname(pdf)}") print(f"make sure json {model} existed and place under {os.path.dirname(pdf)}", file=sys.stderr)
os.exit(1) exit(1)
def read_fn(path): def read_fn(path):
disk_rw = DiskReaderWriter(os.path.dirname(path)) disk_rw = DiskReaderWriter(os.path.dirname(path))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment