Преглед на файлове

fix: reorder backend dropdown initialization in main function

myhloli преди 4 месеца
родител
ревизия
9802015995
променени са 1 файла, в които са добавени 2 реда и са изтрити 13 реда
  1. 2 13
      mineru/cli/gradio_app.py

+ 2 - 13
mineru/cli/gradio_app.py

@@ -175,28 +175,17 @@ def to_pdf(file_path):
 def main():
     example_enable = False
 
-    # try:
-    #     print("预初始化SgLang引擎...")
-    #     from mineru.backend.vlm.vlm_analyze import ModelSingleton
-    #     modelsingleton = ModelSingleton()
-    #     predictor = modelsingleton.get_model("sglang-engine", None, 'http://localhost:30000')
-    #     print("SgLang引擎初始化完成")
-    # except Exception as e:
-    #     print(f"预初始化SgLang引擎失败: {e}")
-
     with gr.Blocks() as demo:
         gr.HTML(header)
         with gr.Row():
             with gr.Column(variant='panel', scale=5):
                 with gr.Row():
                     file = gr.File(label='Please upload a PDF or image', file_types=['.pdf', '.png', '.jpeg', '.jpg'])
-
-                with gr.Row():
-                    backend = gr.Dropdown(["pipeline", "vlm-transformers", "vlm-sglang-client"], label="Backend", value="pipeline")
-
                 with gr.Row():
                     with gr.Column():
                         max_pages = gr.Slider(1, 20, 10, step=1, label='Max convert pages')
+                with gr.Row():
+                    backend = gr.Dropdown(["pipeline", "vlm-transformers", "vlm-sglang-client"], label="Backend", value="pipeline")
                 with gr.Row(visible=True) as ocr_options:
                     with gr.Column():
                         language = gr.Dropdown(all_lang, label='Language', value='ch')