apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): disable HF hub progress

This commit is contained in:
Sean Sube 2023-02-18 09:37:27 -06:00
parent 15060e6c7d
commit 8dd55cc556
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/serve.py
View File

@ -12,6 +12,7 @@ import yaml
from diffusers.utils.logging import disable_progress_bar from diffusers.utils.logging import disable_progress_bar
from flask import Flask, jsonify, make_response, request, send_from_directory, url_for from flask import Flask, jsonify, make_response, request, send_from_directory, url_for
from flask_cors import CORS from flask_cors import CORS
from huggingface_hub.utils.tqdm import disable_progress_bars
from jsonschema import validate from jsonschema import validate
from onnxruntime import get_available_providers from onnxruntime import get_available_providers
from PIL import Image from PIL import Image
@ -391,6 +392,7 @@ load_platforms(context)
if not context.show_progress: if not context.show_progress:
disable_progress_bar() disable_progress_bar()
disable_progress_bars()
app = Flask(__name__) app = Flask(__name__)
CORS(app, origins=context.cors_origin) CORS(app, origins=context.cors_origin)