diff --git a/model_server/scripts/run_server.py b/model_server/scripts/run_server.py index 02b22660d6333d64654f3bf08cff258c55c9d2e3..ca8126e4de0c782009373e078d70a637fbb70b2a 100644 --- a/model_server/scripts/run_server.py +++ b/model_server/scripts/run_server.py @@ -1,6 +1,8 @@ import argparse from multiprocessing import Process import requests +from requests.adapters import HTTPAdapter +from urllib3 import Retry import uvicorn import webbrowser @@ -47,7 +49,13 @@ def main(args) -> None: server_process.start() try: - resp = requests.get(url) + sesh = requests.Session() + retries = Retry( + total=5, + backoff_factor=0.1, + ) + sesh.mount('http://', HTTPAdapter(max_retries=retries)) + resp = sesh.get(url) assert resp.status_code == 200 except Exception: print('Error starting server')