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

blacked

Browse Source
This commit is contained in:
cdietrich 2022-03-02 10:28:32 +01:00
parent 1e5da128f1
commit 8a1df76078
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
image_prediction/predictor.py
View File

@ -27,9 +27,7 @@ class Predictor:
"""
try:
if model_handle is None:
reader = MlflowModelReader(
run_id=CONFIG.service.run_id, mlruns_dir=MLRUNS_DIR
)
reader = MlflowModelReader(run_id=CONFIG.service.run_id, mlruns_dir=MLRUNS_DIR)
self.model_handle = reader.get_model_handle(BASE_WEIGHTS)
else:
self.model_handle = model_handle