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

renaming

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-31 14:52:37 +02:00
parent 4ebb36247e
commit d4604a2cb5
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
image_prediction/default_objects.py
View File

@ -11,7 +11,7 @@ def get_mlflow_model_loader(mlruns_dir):
return model_loader return model_loader
def load_up_pipeline(): def load_pipeline():
model_loader = get_mlflow_model_loader(MLRUNS_DIR) model_loader = get_mlflow_model_loader(MLRUNS_DIR)
model_identifier = CONFIG.service.run_id model_identifier = CONFIG.service.run_id

4
scripts/run_pipeline.py
View File

@ -1,7 +1,7 @@
import argparse import argparse
import json import json
from image_prediction.default_objects import load_up_pipeline from image_prediction.default_objects import load_pipeline
def parse_args(): def parse_args():
@ -14,7 +14,7 @@ def parse_args():
def main(args): def main(args):
pipeline = load_up_pipeline() pipeline = load_pipeline()
with open(args.pdf, "rb") as f: with open(args.pdf, "rb") as f:
predictions = pipeline(f.read()) predictions = pipeline(f.read())