Pārlūkot izejas kodu

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

kernel.h@qq.com 1 gadu atpakaļ
vecāks
revīzija
211590404d
1 mainītis faili ar 3 papildinājumiem un 2 dzēšanām
  1. 3 2
      magic_pdf/cli/magicpdf.py

+ 3 - 2
magic_pdf/cli/magicpdf.py

@@ -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))