mlflow attempt task 2 prediction

2022-05-15 12:57:28 +02:00 · 2022-05-15 12:57:28 +02:00 · e9ebc9a95b
commit e9ebc9a95b
parent 0dd47886dd
1 changed files with 1 additions and 1 deletions
--- a/predict.py
+++ b/predict.py
@ -11,7 +11,7 @@ input = str((sys.argv[1:])[0])
 PATH = "ml/mlruns/1/f65f936936024133a2c03e1e486ba9cf/artifacts/model/"
 model =mlflow.pyfunc.load_model(f"{PATH}")

-with open(f'[PATH]/{input}', 'r') as file:
+with open(f'{PATH}{input}', 'r') as file:
    json_data = json.load(file)

    print(f"Input: {json_data['inputs'][0]}")