Update index.js

index.js

@@ -179,21 +179,18 @@ async function pipeStreamAndReportProgress(readableStream, writableStream) {
   }
 }
 
-/**
- * For non-Chromium browsers, the user must upload the model manually.
- */
-async function loadLlmNonChromium() {
-  alert('Non-Chromium browsers are not supported yet. Please run demo on Chrome for the best experience.');
-  loaderMessage.textContent = "Non-Chromium browsers are not supported yet. Please run on Chrome for best experience.";
-}
-
 /**
  * Loads the LLM file from either cache or OAuth-guarded remote download.
  */
 async function loadLlm() {
-  const isSafari = /^((?!chrome|android).)*safari/i.test(navigator.userAgent);
-  const isFirefox = navigator.userAgent.toLowerCase().indexOf('firefox') > -1;
-  if (isSafari || isFirefox) return loadLlmNonChromium();
+  // For browsers other than Chrome and Edge, we just show an alert and error out.
+  const isChrome = navigator.userAgent.includes('Chrome');
+  const isEdge = navigator.userAgent.includes('Edg');
+  if (!isChrome && !isEdge) {
+    alert('Non-Chromium browsers are not supported yet. Please run demo on Chrome for the best experience.');
+    loaderMessage.textContent = "Non-Chromium browsers are not supported yet. Please run on Chrome for best experience.";
+    return;
+  }
 
   let opfs = await navigator.storage.getDirectory();
   // If we can load the model from cache, then do so.