LLaMA-Factory/src/webui.py
hiyouga f4bf49e891 enable inbrowser in webui
Former-commit-id: 694a05fd044bbbad107ca8fed5494460c78e1981
2024-05-17 00:08:56 +08:00

14 lines
354 B
Python

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()