app.py

# Interface Gradio pour afficher la liste des modèles chargés
import gradio as gr
import asyncio
from db.models import fetch_models_for_group
from models.loader import load_models, model_pipelines
from config.settings import RESOURCE_GROUP

async def init_models():
    """Charger les modèles au démarrage"""
    models_data = await fetch_models_for_group(RESOURCE_GROUP)
    await load_models(models_data)

# Initialisation des modèles
asyncio.run(init_models())

def get_model_list():
    """Retourner la liste des modèles chargés"""
    return list(model_pipelines.keys())

with gr.Blocks() as demo:
    demo.title("Tamis AI - Liste des modèles chargés")
    gr.JSON(get_model_list)

demo.launch()