39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
import argparse
|
||
from pathlib import Path
|
||
|
||
from astrai.inference.server import run_server
|
||
|
||
|
||
def main():
|
||
parser = argparse.ArgumentParser(description="Start AstrAI inference HTTP server")
|
||
parser.add_argument(
|
||
"--host", default="0.0.0.0", help="Host address (default: 0.0.0.0)"
|
||
)
|
||
parser.add_argument(
|
||
"--port", type=int, default=8000, help="Port number (default: 8000)"
|
||
)
|
||
parser.add_argument(
|
||
"--reload", action="store_true", help="Enable auto‑reload for development"
|
||
)
|
||
parser.add_argument(
|
||
"--param-path",
|
||
type=Path,
|
||
default=None,
|
||
help="Path to model parameters (default: project_root/params)",
|
||
)
|
||
args = parser.parse_args()
|
||
|
||
# If param_path is provided, set environment variable or modify global?
|
||
# Currently the server loads from default location on startup.
|
||
# We could pass it via an environment variable, but for simplicity we assume
|
||
# the default location is correct.
|
||
project_root = Path(__file__).parent.parent
|
||
param_path = args.param_path or (project_root / "params")
|
||
print(f"Starting AstrAI inference server on http://{args.host}:{args.port}")
|
||
print(f"Model parameters expected at: {[param_path]}")
|
||
run_server(host=args.host, port=args.port, reload=args.reload)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|