Upload folder using huggingface_hub

server/app.py

@@ -63,23 +63,22 @@ def main(host: str = "0.0.0.0", port: int = 8000):
         uv run --project . server --port 8001
         python -m token_optimiser.server.app
 
-    Args:
-        host: Host address to bind to (default: "0.0.0.0")
-        port: Port number to listen on (default: 8000)
-
     For production deployments, consider using uvicorn directly with
     multiple workers:
         uvicorn token_optimiser.server.app:app --workers 4
     """
+    import argparse
     import uvicorn
 
-    uvicorn.run(app, host=host, port=port)
-
+    if host == "0.0.0.0" and port == 8000:
+        parser = argparse.ArgumentParser()
+        parser.add_argument("--host", type=str, default="0.0.0.0")
+        parser.add_argument("--port", type=int, default=8000)
+        args = parser.parse_args()
+        host = args.host
+        port = args.port
 
-if __name__ == '__main__':
-    import argparse
+    uvicorn.run(app, host=host, port=port)
 
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--port", type=int, default=8000)
-    args = parser.parse_args()
-    main(port=args.port)
+if __name__ == "__main__":
+    main()