diff --git a/utils/flask_rest_api/restapi.py b/utils/flask_rest_api/restapi.py index b93ad16a0f58..38868cc98d84 100644 --- a/utils/flask_rest_api/restapi.py +++ b/utils/flask_rest_api/restapi.py @@ -1,5 +1,5 @@ """ -Run a rest API exposing the yolov5s object detection model +Run a Flask REST API exposing a YOLOv5s model """ import argparse import io @@ -31,7 +31,10 @@ def predict(): if __name__ == "__main__": parser = argparse.ArgumentParser(description="Flask API exposing YOLOv5 model") parser.add_argument("--port", default=5000, type=int, help="port number") - args = parser.parse_args() + opt = parser.parse_args() + + # Fix known issue urllib.error.HTTPError 403: rate limit exceeded https://github.com/ultralytics/yolov5/pull/7210 + torch.hub._validate_not_a_forked_repo = lambda a, b, c: True model = torch.hub.load("ultralytics/yolov5", "yolov5s", force_reload=True) # force_reload to recache - app.run(host="0.0.0.0", port=args.port) # debug=True causes Restarting with stat + app.run(host="0.0.0.0", port=opt.port) # debug=True causes Restarting with stat