Update index.html

index.html

@@ -161,7 +161,7 @@
                     <option value="quality">Highest Quality</option>
                 </select>
                 <div id="model-info">
-                    TTS: Xenova/mms-tts-eng / STT: Xenova/whisper-tiny.en / LLM: Xenova/gpt2-tiny-english
+                    TTS: Xenova/mms-tts-eng / STT: Xenova/whisper-tiny.en / LLM: Xenova/Qwen1.5-0.5B-Chat
                 </div>
             </div>
             <div id="visualizer"></div>
@@ -237,10 +237,10 @@
                 [sttPipeline, ttsPipeline, llmPipeline] = await Promise.all([
                     pipeline('automatic-speech-recognition', 'Xenova/whisper-tiny.en', { quantized: true }),
                     pipeline('text-to-speech', 'Xenova/mms-tts-eng', { quantized: true }),
-                    pipeline('text-generation', 'Xenova/gpt2-tiny-english', { quantized: true })
+                    pipeline('text-generation', 'Xenova/Qwen1.5-0.5B-Chat', { quantized: true })
                 ]);
               
-                addLog('System: Digital Human Voice Chat initialized with GPT-2 Tiny. Click "Begin Call" to start.');
+                addLog('System: Digital Human Voice Chat initialized with Qwen1.5-0.5B-Chat. Click "Begin Call" to start.');
                 startButton.disabled = false;
                 loadingDiv.style.display = 'none';
             } catch (error) {
@@ -259,11 +259,17 @@
                 const transcription = await sttPipeline(audio);
                 addLog(`User: ${transcription.text}`);
                 
-                const llmResponse = await llmPipeline(transcription.text, {
-                    max_new_tokens: 50,
-                    temperature: 0.7
+                const messages = [
+                    { role: 'system', content: 'You are a helpful assistant.' },
+                    { role: 'user', content: transcription.text }
+                ];
+
+                const llmResponse = await llmPipeline(messages, {
+                    max_new_tokens: 128,
+                    do_sample: false
                 });
-                const botResponse = llmResponse[0].generated_text;
+
+                const botResponse = llmResponse[0].generated_text.find(msg => msg.role === 'assistant').content;
                 addLog(`Bot: ${botResponse}`);
 
                 isSpeaking = true;