Update index.html

index.html

@@ -26,14 +26,22 @@
         env.allowLocalModels = false;
 
         // Reconhecimento de Entidades Nomeadas (NER)
-        const nerStatus = document.getElementById('status');
         const textInput = document.getElementById('text-input');
         const analyzeTextButton = document.getElementById('analyze-text');
         const textOutput = document.getElementById('text-output');
 
-        nerStatus.textContent = 'Carregando modelo de NER...';
-        const nerModel = await pipeline('ner', 'Xenova/distilbert-base-multilingual-cased-ner-hrl');
-        nerStatus.textContent = 'Modelo de NER pronto!';
+        // Carregar modelo NER com quantização q4
+        let nerModel;
+        try {
+            textOutput.textContent = 'Carregando modelo de NER...';
+            nerModel = await pipeline('ner', 'Xenova/distilbert-base-multilingual-cased-ner-hrl', {
+                dtype: 'q4'
+            });
+            textOutput.textContent = 'Modelo de NER pronto!';
+        } catch (error) {
+            console.error('Erro ao carregar o modelo:', error);
+            textOutput.textContent = 'Erro ao carregar o modelo. Tente novamente mais tarde.';
+        }
 
         analyzeTextButton.addEventListener('click', async () => {
             const inputText = textInput.value.trim();
@@ -43,8 +51,13 @@
             }
 
             textOutput.textContent = 'Analisando...';
-            const nerOutput = await nerModel(inputText);
-            renderEntities(nerOutput);
+            try {
+                const nerOutput = await nerModel(inputText);
+                renderEntities(nerOutput);
+            } catch (error) {
+                console.error('Erro durante a análise:', error);
+                textOutput.textContent = 'Erro durante a análise. Verifique o texto inserido.';
+            }
         });
 
         function renderEntities(entities) {