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
import os
import json as json_parse
import sys
import click
from loguru import logger
from pathlib import Path
......@@ -163,8 +164,8 @@ def pdf_command(pdf, model, method):
if model is None:
model = pdf.replace(".pdf", ".json")
if not os.path.exists(model):
print(f"make sure json file existed and place under {os.dirname(pdf)}")
os.exit(1)
print(f"make sure json {model} existed and place under {os.path.dirname(pdf)}", file=sys.stderr)
exit(1)
def read_fn(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