Update app.py

app.py

@@ -1,5 +1,4 @@
 
-
 import os, shlex, subprocess, tempfile, traceback, time, glob, gc, shutil
 import torch
 from huggingface_hub import snapshot_download
@@ -58,30 +57,27 @@ def get_model(name):
     
     if not nemo_file: raise FileNotFoundError("Fichier .nemo introuvable.")
 
-    # Correctif pour les clés "embedding_model" inattendues et erreur __init__
-    try:
-        model = nemo_asr.models.ASRModel.restore_from(
-            nemo_file, 
-            map_location=torch.device(DEVICE),
-            strict=False,
-            override_config_path=None
-        )
-    except Exception as e:
-        print(f"⚠️ Tentative de chargement alternatif : {e}")
-        # Fallback sans override_config_path
-        model = nemo_asr.models.ASRModel.restore_from(
-            nemo_file, 
-            map_location=torch.device(DEVICE),
-            strict=False
-        )
+    # Méthode la plus simple et robuste pour charger le modèle
+    print(f"📦 Chargement du modèle depuis : {nemo_file}")
     
-    model.to(DEVICE).eval()
+    # On laisse NeMo gérer tout automatiquement
+    model = nemo_asr.models.ASRModel.restore_from(nemo_file)
+    
+    # Déplacement vers le device approprié
+    model = model.to(DEVICE)
+    model.eval()
+    
+    # Optimisation mémoire pour GPU
     if DEVICE == "cuda":
-        model.half()
+        try:
+            model = model.half()
+        except Exception as e:
+            print(f"⚠️ Impossible de convertir en half precision: {e}")
         
     _cache[name] = model
     return model
 
+
 # 3. UTILITAIRES
 def format_srt_time(sec):
     td = time.gmtime(sec)