UNPKG

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
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!")