import json
import mlflow
import numpy as np
import sys
import tarfile

file = tarfile.open('mlruns.tar.gz')
file.extractall('./ml')

input = str((sys.argv[1:])[0])
PATH = "ml/mlruns/1/f65f936936024133a2c03e1e486ba9cf/artifacts/model/"
model = mlflow.pytorch.load_model(f"{PATH}/MLmodel")

with open(f'[PATH]/{input}', 'r') as file:
    json_data = json.load(file)
    
    print(f"Input: {json_data['inputs'][0]}")
    print(f"Prediction: {model.predict(np.array([json_data['inputs'][0]], dtype=np.float32))}")