teachable-machine.js
Version:
A robust and optimized JavaScript library for integrating Google's Teachable Machine models, supporting various image sources and providing efficient classification capabilities.
24 lines (18 loc) • 847 B
JavaScript
import teachablemachine from './index.js';
const MODEL_URL = "https://teachablemachine.withgoogle.com/models/r6BBk-hiN/";
const IMAGE_URL = "https://media-blog.sashido.io/content/images/2020/09/SashiDo_Dog.jpg";
console.log("Loading model...");
const model = await teachablemachine.create({ modelUrl: MODEL_URL });
console.log("Model loaded successfully!");
console.log("-----------------------------------------");
console.log("Running model...");
const start = Date.now();
const predictions = await model.classify({ imageUrl: IMAGE_URL });
const end = Date.now();
console.log("Time:", end - start, "ms")
console.log("Classification results:");
console.log(predictions)
// Reset model data
console.log("Classification is finished. Resetting model data...")
model.dispose();
console.log("Model resetted successfully!")