| import os | |
| import subprocess | |
| import sys | |
| def on_start(): | |
| print("Executing './on_start.sh' (downloading models, etc.)") | |
| result = subprocess.run("./on_start.sh", shell=True, env=os.environ) | |
| if result.returncode != 0: | |
| raise RuntimeError(f"Error executing ./on_start.sh [exit code: {result.returncode}]") | |
| def start(): | |
| on_start() | |
| print(f"Launching {'API server' if '--nowebui' in sys.argv else 'Web UI'} with arguments: {' '.join(sys.argv[1:])}") | |
| import webui # noqa | |
| if '--nowebui' in sys.argv: | |
| webui.api_only() | |
| else: | |
| webui.webui() | |
| if __name__ == "__main__": | |
| start() |