add more docs

getreflect · Apr 7, 2020 · b654b92 · b654b92
1 parent f18b5d9
commit b654b92
Show file tree

Hide file tree

Showing 8 changed files with 10 additions and 19 deletions.
diff --git a/dist/src/background.js b/dist/src/background.js
diff --git a/dist/src/background.js.map b/dist/src/background.js.map
diff --git a/dist/src/content.js b/dist/src/content.js
@@ -83,7 +83,6 @@ function callBackgroundWithIntent(intent) {
     // open connection to runtime (background.ts)
     const port = chrome.runtime.connect({ name: "intentStatus" });
     port.postMessage({ intent: intent, url: window.location.href });
-    displayStatus("connecting...", 3000, REFLECT_INFO);
     port.onMessage.addListener((msg) => {
         switch (msg.status) {
             case "ok":
@@ -99,10 +98,6 @@ function callBackgroundWithIntent(intent) {
                 // clear input
                 $("#textbox").val("");
                 break;
-            case "timeout":
-                // display message
-                displayStatus("couldn't reach server, try again later.", 3000, REFLECT_ERR);
-                break;
         }
         // close connection
         port.disconnect();

diff --git a/dist/src/nn.js b/dist/src/nn.js
diff --git a/dist/src/nn.js.map b/dist/src/nn.js.map
diff --git a/src/background.ts b/src/background.ts
@@ -110,7 +110,6 @@ chrome.runtime.onConnect.addListener((port) => {
 		const url: string = msg.url;
 
 		const valid: boolean = await model.predict(intent);
-		console.log(valid)
 
 		if (valid) {
 			// add whitelist period for site

diff --git a/src/content.ts b/src/content.ts
@@ -101,9 +101,6 @@ function callBackgroundWithIntent(intent: string): void {
     // open connection to runtime (background.ts)
     const port = chrome.runtime.connect({name: "intentStatus"});
 	port.postMessage({intent: intent, url: window.location.href});
-
-	displayStatus("connecting...", 3000, REFLECT_INFO);
-
 	port.onMessage.addListener((msg) => {
 		switch (msg.status) {
 			case "ok":
@@ -122,11 +119,6 @@ function callBackgroundWithIntent(intent: string): void {
 				// clear input
 				$("#textbox").val("");
 				break;
-
-			case "timeout":
-				// display message
-				displayStatus("couldn't reach server, try again later.", 3000, REFLECT_ERR);
-				break;
 		}
 
 		// close connection

diff --git a/src/nn.ts b/src/nn.ts
@@ -85,6 +85,11 @@ export class IntentClassifier {
 
 		// predict
 		return (this.model.predict(inputTensor) as tf.Tensor).data().then(predictions => {
+
+			// garbage collect finished tensor to prevent mem leak
+			tf.dispose(inputTensor)
+
+			// threshold net output
 			const confidence: number = predictions[0];
 			if (confidence > threshold) {
 				return true;