Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,6 @@ OUTPUT_DIR = "output_results"
|
|
| 11 |
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
| 12 |
|
| 13 |
# --- 模型加载器 ---
|
| 14 |
-
# 这个函数会在GPU会话中被调用
|
| 15 |
def load_gpu_model():
|
| 16 |
print("正在加载经过版本锁定的PaddleOCR GPU模型...")
|
| 17 |
# 使用与锁定版本兼容的参数：use_gpu=True是必需的
|
|
@@ -29,7 +28,7 @@ def process_pdf_max_speed(pdf_file, progress=gr.Progress(track_tqdm=True)):
|
|
| 29 |
return "请先上传一个PDF文件。", None
|
| 30 |
|
| 31 |
try:
|
| 32 |
-
# 在GPU
|
| 33 |
ocr = load_gpu_model()
|
| 34 |
|
| 35 |
doc = fitz.open(pdf_file.name)
|
|
|
|
| 11 |
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
| 12 |
|
| 13 |
# --- 模型加载器 ---
|
|
|
|
| 14 |
def load_gpu_model():
|
| 15 |
print("正在加载经过版本锁定的PaddleOCR GPU模型...")
|
| 16 |
# 使用与锁定版本兼容的参数：use_gpu=True是必需的
|
|
|
|
| 28 |
return "请先上传一个PDF文件。", None
|
| 29 |
|
| 30 |
try:
|
| 31 |
+
# 在GPU会话中加载模型
|
| 32 |
ocr = load_gpu_model()
|
| 33 |
|
| 34 |
doc = fitz.open(pdf_file.name)
|