Merge pull request #1157 from mofanke/main

Update darknet.js
ml5js · Jan 21, 2022 · ce2f9f5 · ce2f9f5
2 parents 54e4fd4 + 04345bf
commit ce2f9f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/ImageClassifier/darknet.js b/src/ImageClassifier/darknet.js
@@ -93,7 +93,8 @@ export class Darknet {
   }
 }
 
-export async function load(version) {
+export async function load(modelConfig) {
+  const { version } = modelConfig
   if (version !== "reference" && version !== "tiny") {
     throw new Error("Please select a version: darknet-reference or darknet-tiny");
   }