fix: got_ocr

app.py

@@ -74,7 +74,7 @@ def perform_ocr(selected_index, image_paths):
     if selected_index is not None and image_paths and 0 <= selected_index < len(image_paths):
         selected_image = image_paths[selected_index][0]
         # 这里添加OCR处理逻辑
-        return got_ocr(model, selected_image)
+        return got_ocr(model, tokenizer, selected_image)
     return "请先选择一个图片"
 
 

got_ocr.py

@@ -2,7 +2,7 @@ import base64
 import os
 
 
-def got_ocr(model, image_path, got_mode="plain texts OCR", fine_grained_mode="", ocr_color="", ocr_box=""):
+def got_ocr(model, tokenizer, image_path, got_mode="plain texts OCR", fine_grained_mode="", ocr_color="", ocr_box=""):
     # 执行OCR
     try:
         if got_mode == "plain texts OCR":
@@ -47,7 +47,7 @@ if __name__ == "__main__":
     model = model.eval().cuda()
 
     image_path = "path/to/your/image.png"
-    result, html = got_ocr(model, image_path, got_mode="format texts OCR")
+    result, html = got_ocr(model, tokenizer, image_path, got_mode="format texts OCR")
     print("OCR结果:", result)
     if html:
         print("HTML结果可用")