Update model lookup to use filename instead of ID in get_model function

models/loader.py

@@ -127,11 +127,11 @@ async def load_models(models_data: List[Dict[str, Any]]) -> None:
     # return model_pipelines # Ancienne logique
 
 
-def get_model(model_id: str):
-    """Récupérer un modèle chargé par son ID.
+def get_model(model_name: str):
+    """Récupérer un modèle chargé par son nom de fichier (hf_filename).
     
     Args:
-        model_id: ID du modèle à récupérer
+        model_name: Nom du fichier du modèle (hf_filename) à récupérer
         
     Returns:
         Le modèle chargé (le pipeline)
@@ -139,9 +139,13 @@ def get_model(model_id: str):
     Raises:
         KeyError: Si le modèle n'est pas trouvé
     """
-    if model_id not in model_pipelines:
-        logger.error(f"Model {model_id} not found in loaded models")
-        raise KeyError(f"Model {model_id} not found")
+    # Rechercher par hf_filename
+    for model_id, model_data in model_pipelines.items():
+        metadata = model_data.get('metadata', {})
+        if metadata.get('hf_filename') == model_name:
+            logger.info(f"Model found: {model_name} (ID: {model_id})")
+            return model_data['pipeline']
     
-    # Retourner uniquement le pipeline depuis le dictionnaire stocké
-    return model_pipelines[model_id]['pipeline']
+    # Si on arrive ici, le modèle n'a pas été trouvé
+    logger.error(f"Model {model_name} not found in loaded models")
+    raise KeyError(f"Model {model_name} not found")