Update index.js

index.js

@@ -170,11 +170,11 @@ async function pipeStreamAndReportProgress(readableStream, writableStream) {
 }
 
 /**
- * For Safari, the user must upload the model manually.
+ * For non-Chromium browsers, the user must upload the model manually.
  */
-async function loadLlmSafari() {
-  alert('Safari browser does not support automatic model downloading. However, you can try downloading the model manually and then selecting it from the file chooser.');
-  loaderMessage.textContent = "Safari not supported. You can try downloading the model yourself manually, and then selecting it from the file chooser.";
+async function loadLlmNonChromium() {
+  alert('Your browser does not support automatic model downloading. Please run demo on Chrome, or you can try downloading the model manually and then selecting it from the file chooser.');
+  loaderMessage.textContent = "Non-Chrome browsers are not fully supported. You can try downloading the model yourself manually, and then selecting it from the file chooser.";
   const fileButton = document.getElementById('safari-upload-file-button');
   const fileChooser = document.getElementById('safari-upload-file-chooser');
   fileChooser.style = "visibility:hidden";
@@ -197,8 +197,9 @@ async function loadLlmSafari() {
  */
 async function loadLlm() {
   const isSafari = /^((?!chrome|android).)*safari/i.test(navigator.userAgent);
-  if (isSafari) return loadLlmSafari();
-  
+  const isFirefox = navigator.userAgent.toLowerCase().indexOf('firefox') > -1;
+  if (isSafari || isFirefox) return loadLlmNonChromium();
+
   let opfs = await navigator.storage.getDirectory();
   // If we can load the model from cache, then do so.
   try {