diff --git a/web/packages/new/photos/services/ml/embedding.ts b/web/packages/new/photos/services/ml/embedding.ts index f5b1ab0bc3..beb80309eb 100644 --- a/web/packages/new/photos/services/ml/embedding.ts +++ b/web/packages/new/photos/services/ml/embedding.ts @@ -46,7 +46,9 @@ import { type RemoteFaceIndex } from "./face"; * all the derived data associated with a file. */ // TODO-ML: Fix name to "combined" before release -type EmbeddingModel = "xxxx-xxxx" /* Combined format */; +const wipModelName = process.env.NEXT_PUBLIC_ENTE_ENABLE_WIP_ML_DONT_USE ?? ""; +// type EmbeddingModel = "xxxx-xxxx" /* Combined format */; +type EmbeddingModel = string; // "xxxx-xxxx" /* Combined format */; const RemoteEmbedding = z.object({ /** The ID of the file whose embedding this is. */ @@ -200,7 +202,7 @@ export const fetchDerivedData = async ( filesByID: Map, ): Promise> => { // TODO-ML: Fix name to "combined" before release - const remoteEmbeddings = await fetchEmbeddings("xxxx-xxxx", [ + const remoteEmbeddings = await fetchEmbeddings(wipModelName, [ ...filesByID.keys(), ]); @@ -296,7 +298,7 @@ export const putDerivedData = async ( // TODO-ML: Fix name to "combined" before release putEmbedding( enteFile, - "xxxx-xxxx", + wipModelName, await gzip(JSON.stringify(derivedData)), ); diff --git a/web/packages/new/photos/services/ml/index.ts b/web/packages/new/photos/services/ml/index.ts index 1b019bd4cf..e09bc7484d 100644 --- a/web/packages/new/photos/services/ml/index.ts +++ b/web/packages/new/photos/services/ml/index.ts @@ -99,7 +99,7 @@ export const terminateMLWorker = () => { */ // TODO-ML: export const isMLSupported = - isDesktop && process.env.NEXT_PUBLIC_ENTE_ENABLE_WIP_ML; + isDesktop && process.env.NEXT_PUBLIC_ENTE_ENABLE_WIP_ML_DONT_USE; /** * Was this someone who might've enabled the beta ML? If so, show them the