Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

logger refactoring

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-16 14:00:39 +01:00
parent 0e004cbd21
commit d76fefd3ff
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
config.yaml
View File

@ -26,4 +26,3 @@ filters:
max: $MAX_IMAGE_FORMAT|10 # Maximum permissible
min_confidence: $MIN_CONFIDENCE|0.5 # Minimum permissible prediction confidence

1
image_prediction/flask.py
View File

@ -6,6 +6,7 @@ from image_prediction.utils import get_logger
logger = get_logger()
def make_prediction_server(predict_fn: Callable):
app = Flask(__name__)

6
src/serve.py
View File

@ -6,15 +6,15 @@ from image_prediction.config import CONFIG
from image_prediction.flask import make_prediction_server
from image_prediction.predictor import Predictor
from image_prediction.response import build_response
from image_prediction.utils import get_logger
logger = logging.getLogger(__name__)
logger.setLevel(CONFIG.service.logging_level)
logger = get_logger()
def main():
def predict(pdf):
predictions, metadata = predictor.predict_pdf(pdf)
predictions, metadata = predictor.predict_pdf(pdf, verbose=CONFIG.service.progressbar)
response = build_response(predictions, metadata)
return response