Update app.py

app.py

@@ -3,14 +3,15 @@ import torch
 import torchaudio
 from torchaudio.transforms import Resample
 
-# Ensure that wav2vec2.py is imported only once
-try:
+# Function to import wav2vec2 model, avoiding duplicate registration
+def import_wav2vec2():
+    import sys
+    if 'wav2vec2' not in sys.modules:
+        import wav2vec2
     from wav2vec2 import Wav2Vec2Model, Wav2Vec2Config
-except ValueError as e:
-    if "Cannot register duplicate model" in str(e):
-        pass
-    else:
-        raise e
+    return Wav2Vec2Model, Wav2Vec2Config
+
+Wav2Vec2Model, Wav2Vec2Config = import_wav2vec2()
 
 # 定义模型路径
 model_path = "https://huggingface.co/Tele-AI/TeleSpeech-ASR1.0/resolve/main/finetune_large_kespeech.pt"