Quellcode durchsuchen

feat: add server name and port options to Gradio app for enhanced configuration

myhloli vor 4 Monaten
Ursprung
Commit
9a16d3c74f
1 geänderte Dateien mit 19 neuen und 2 gelöschten Zeilen
  1. 19 2
      mineru/cli/gradio_app.py

+ 19 - 2
mineru/cli/gradio_app.py

@@ -232,7 +232,24 @@ def update_interface(backend_choice):
     help="Set the maximum number of pages to convert from PDF to Markdown.",
     default=1000,
 )
-def main(example_enable, sglang_engine_enable, mem_fraction_static, torch_compile_enable, api_enable, max_convert_pages):
+@click.option(
+    '--server-name',
+    'server_name',
+    type=str,
+    help="Set the server name for the Gradio app.",
+    default=None,
+)
+@click.option(
+    '--server-port',
+    'server_port',
+    type=int,
+    help="Set the server port for the Gradio app.",
+    default=None,
+)
+def main(
+        example_enable, sglang_engine_enable, mem_fraction_static, torch_compile_enable, api_enable, max_convert_pages,
+        server_name, server_port
+):
     if sglang_engine_enable:
         try:
             print("Start init SgLang engine...")
@@ -336,7 +353,7 @@ def main(example_enable, sglang_engine_enable, mem_fraction_static, torch_compil
             api_name=api_name
         )
 
-    demo.launch()
+    demo.launch(server_name=server_name, server_port=server_port, show_api=api_enable)
 
 
 if __name__ == '__main__':