Update app.py

app.py

@@ -29,7 +29,7 @@ DEMO_PATH = os.getenv("DEMO_PATH", "./demo")
 TMP_PATH = os.getenv("TMP_PATH", "./demo/temp")
 MODELS_PATH = os.getenv("MODELS_PATH", "./pretrained_models")
 device = "cuda" if torch.cuda.is_available() else "cpu"
-whisper_model, align_model, ssrspeech_model = None, None, None
+transcribe_model, align_model, ssrspeech_model = None, None, None
 
 def get_random_string():
     return "".join(str(uuid.uuid4()).split("-"))
@@ -134,16 +134,16 @@ def load_models(ssrspeech_model_name):
         ssrspeech_model_name = "English"
         text_tokenizer = TextTokenizer(backend="espeak")
         language = "en"
-        whisper_model_name = "base.en"
+        transcribe_model_name = "base.en"
         
     elif ssrspeech_model_name == "Mandarin":
         ssrspeech_model_name = "Mandarin"
         text_tokenizer = TextTokenizer(backend="espeak", language='cmn')
         language = "zh"
-        whisper_model_name = "base"
+        transcribe_model_name = "base"
 
     align_model = WhisperxAlignModel(language)
-    transcribe_model = WhisperxModel(whisper_model_name, align_model, language)
+    transcribe_model = WhisperxModel(transcribe_model_name, align_model, language)
 
     ssrspeech_fn = f"{MODELS_PATH}/{ssrspeech_model_name}.pth"
     if not os.path.exists(ssrspeech_fn):