const i = require("./lib.js");

const nn = new i.NeuralNet();

async function start() {
  await nn.init();
  await nn.trainDir("./photos/cats/", 0);
  console.log("Dogs: ")
  await nn.trainDir("./photos/dogs/", 1);
};

async function predict(path) {
  let pred = await nn.predict(nn.image(path));
  console.log(pred)
}