From 8dd55cc5567842cd95ebbb935fcaaaa1affd108c Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 18 Feb 2023 09:37:27 -0600 Subject: [PATCH] fix(api): disable HF hub progress --- api/onnx_web/serve.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index 3e75c041..9e492e19 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -12,6 +12,7 @@ import yaml from diffusers.utils.logging import disable_progress_bar from flask import Flask, jsonify, make_response, request, send_from_directory, url_for from flask_cors import CORS +from huggingface_hub.utils.tqdm import disable_progress_bars from jsonschema import validate from onnxruntime import get_available_providers from PIL import Image @@ -391,6 +392,7 @@ load_platforms(context) if not context.show_progress: disable_progress_bar() + disable_progress_bars() app = Flask(__name__) CORS(app, origins=context.cors_origin)