diff --git a/src/utils/machineLearning/compatibility.ts b/src/utils/machineLearning/compatibility.ts
index b488065e7..d18e6ffdd 100644
--- a/src/utils/machineLearning/compatibility.ts
+++ b/src/utils/machineLearning/compatibility.ts
@@ -4,9 +4,12 @@ import {
     webglSupported,
 } from 'utils/common';
 
+import isElectron from 'is-electron';
+
 // TODO: check electron env to be campatible with this
 export function canEnableMlSearch(): boolean {
-    if (!runningInChrome(false)) {
+    // check if is chrome or ente desktop
+    if (!runningInChrome(false) && !isElectron()) {
         console.log('Not running in Chrome Desktop');
         return false;
     }