Update app.py

app.py

@@ -10,7 +10,7 @@ from transformers import pipeline  # only pipeline is needed for TTS
 
 # ---------------- CONFIG ---------------- #
 MODEL_NAME = "large-v2"
-DEVICE = "cpu"
+DEVICE = "cpu"  # change to "cuda" if GPU available in Space
 
 LANG_CODES = {
     "English": "en", "Tamil": "ta", "Malayalam": "ml",
@@ -145,7 +145,7 @@ def compare_pronunciation(audio, language_choice, intended_sentence,
     actual_text = transcribe_once(audio, lang_code, primer_weak,
                                   pass1_beam, pass1_temp, pass1_condition)
 
-    # Pass 2 (fixed)
+    # Pass 2 (fixed best-known defaults)
     strict_prompt = f"{primer_strong}\nTarget: {intended_sentence}"
     corrected_text = transcribe_once(audio, lang_code, strict_prompt,
                                      beam_size=5, temperature=0.0, condition_on_previous_text=False)
@@ -163,7 +163,7 @@ def compare_pronunciation(audio, language_choice, intended_sentence,
     diff_html = highlight_differences(intended_sentence, actual_text)
     char_html = char_level_highlight(intended_sentence, actual_text)
 
-    # TTS for intended & pass1
+    # TTS
     tts_intended = synthesize_tts(intended_sentence, language_choice)
     tts_pass1 = synthesize_tts(actual_text, language_choice)