Ai to predict a sequence of 6 numbers of a file with 3631 sequences

def predict(file_path):
  model = load_model('model_weights.h5')
  data = pd.read_csv(file_path, header = None)
  data = data.drop(data.columns[0], axis=1)
  data = data.values
  data = data.reshape((data.shape[0], data.shape[1], 1))
  
  prediction = model.predict(data)
  prediction = prediction.reshape((prediction.shape[0], prediction.shape[1]))
  
  return prediction