ium_434742/predict-s434704.py

13 lines
269 B
Python
Raw Normal View History

import json
import sys
import mlflow
filename = sys.argv[1]
model = mlflow.keras.load_model("movies_on_streaming_platforms_model")
path = 'movies_on_streaming_platforms_model/' + filename
with open(path) as f:
data = json.load(f)
model.predict(data['inputs'])