Src/Inference.py

## Start here with the inference procedure
import torch
from Models_Class.LSTMModel import LSTMModel

def load_model(model_path, input_size, hidden_size, output_size, num_layers):
    loaded_model = LSTMModel(input_size, hidden_size, output_size, num_layers)
    loaded_model.load_state_dict(torch.load(model_path))
    return loaded_model