from joblib import  load
from numpy import array
model=load(r"/home/hemanthvsr/htdocs/forest-hackathon/lstm_model.joblib")


def predict(input_data):
  temp=model.predict(array(input_data).reshape(1, 1, -1))[0][0]
  if temp<0:
    return temp*-1
  return temp

if __name__ == "__main__":
  sample_input = [23, 90, 0.1]
  prediction = predict(sample_input)
  print("Prediction:", prediction)