Merge branch 'now-model' of https://github.com/Mattk70/Chirpity-Electron

into now-model
Mattk70 · Nov 12, 2023 · e01ad1c · e01ad1c
2 parents 3c4769a + 94971ea
commit e01ad1c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/js/model.js b/js/model.js
@@ -101,7 +101,7 @@ onmessage = async (e) => {
                 const spec_width = e.data.width;
                 let image;
                 const signal = tf.tensor1d(buffer, 'float32');
-                const bufferTensor = myModel.normalise_audio(signal);
+                const bufferTensor = myModel.normalise_audio_batch(signal);
                 signal.dispose();
                 const imageTensor = tf.tidy(() => {
                     return myModel.makeSpectrogram(bufferTensor);

diff --git a/js/worker.js b/js/worker.js
@@ -22,7 +22,7 @@ let SEEN_LIST_UPDATE = false // Prevents  list updates from every worker on ever
 
 const DEBUG = false;
 
-const DATASET = false;
+const DATASET = true;
 const adding_chirpity_additions = true;
 const dataset_database = DATASET;
 
@@ -1304,7 +1304,7 @@ const convertSpecsFromExistingSpecs = async (path) => {
 }
 
 const saveResults2DataSet = (rootDirectory) => {
-    if (!rootDirectory) rootDirectory = '/mnt/608E21D98E21A88C/Users/simpo/PycharmProjects/Data/test';
+    if (!rootDirectory) rootDirectory = 'C:/Users/simpo/PycharmProjects/Data/test';
     const height = 256, width = 384;
     let t0 = Date.now()
     let promise = Promise.resolve();