Skip to content
Snippets Groups Projects

TupleTool Search Functionality

Merged Franz Ludwig Kramer requested to merge ttoolsearch into master
All threads resolved!
Files
3
@@ -10,7 +10,7 @@
@@ -10,7 +10,7 @@
\*****************************************************************************/
\*****************************************************************************/
import {env} from "@xenova/transformers";
import {env} from "@xenova/transformers";
// To always download the model from huggingface.co
// To always download the model from huggingface.com
env.allowLocalModels = false;
env.allowLocalModels = false;
env.useBrowserCache = false;
env.useBrowserCache = false;
// Create a new worker
// Create a new worker
@@ -26,7 +26,6 @@ let loadResolve;
@@ -26,7 +26,6 @@ let loadResolve;
* @type {function}
* @type {function}
*/
*/
let queryResolve;
let queryResolve;
let top5;
worker.onmessage = function (event) {
worker.onmessage = function (event) {
const message = event.data;
const message = event.data;
@@ -34,17 +33,13 @@ worker.onmessage = function (event) {
@@ -34,17 +33,13 @@ worker.onmessage = function (event) {
case "progress":
case "progress":
if (message.progress.status === "ready") {
if (message.progress.status === "ready") {
loadResolve();
loadResolve();
} else {
console.log(message.progress.progress);
}
}
break;
break;
case "corpus":
case "corpus":
// the corpus is embedded
// the corpus is embedded
queryResolve();
queryResolve();
//query_embedding = message.data;
break;
break;
case "result":
case "result":
top5 = message.result;
queryResolve(message.result);
queryResolve(message.result);
break;
break;
}
}
@@ -62,8 +57,7 @@ export async function loadModel(modelname) {
@@ -62,8 +57,7 @@ export async function loadModel(modelname) {
});
});
}
}
/**
/**
* Calculate corpus embedding
* Passes corpus embedding from backend to the worker
* Might become obsolete when this is done in the backend
* @returns
* @returns
*/
*/
export async function embedCorpus(metadata) {
export async function embedCorpus(metadata) {
Loading