Files
local-embedding-api/main.py
2024-01-15 12:36:42 +08:00

15 lines
411 B
Python

import argparse
import uvicorn
args = argparse.ArgumentParser()
args.add_argument("--port", type=int, default=7999)
args.add_argument("--host", type=str, default="0.0.0.0")
args.add_argument("--reload", action="store_true")
args = args.parse_args()
if __name__ == "__main__":
print("Start serving on", args.host, args.port)
uvicorn.run("app:app", host=args.host, port=args.port, reload=args.reload)