Upload New File

a92ad34e · zhougaofeng · d6ab59c8 · a92ad34e
Commit a92ad34e authored Jun 11, 2024 by zhougaofeng
Hide whitespace changes
Inline Side-by-side

Showing with 13 additions and 0 deletions

src/webui.py src/webui.py +13 -0

No files found.
--- a/src/webui.py
+++ b/src/webui.py
+import os
+
+from llamafactory.webui.interface import create_ui
+
+
+def main():
+    gradio_share = os.environ.get("GRADIO_SHARE", "0").lower() in ["true", "1"]
+    server_name = os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0")
+    create_ui().queue().launch(share=gradio_share, server_name=server_name, inbrowser=True)
+
+
+if __name__ == "__main__":
+    main()