import mlflow
import sys
import json

model = mlflow.keras.load_model("movies_on_streaming_platforms_model")
with open('movies_on_streaming_platforms_model/input_example.json') as json_file:
    data = json.load(json_file)
print(data)
print(model.predict(data))