alexfremont commited on
Commit
1445bc9
·
1 Parent(s): df3bf97

Improve model info formatting for cleaner display in API responses

Browse files
Files changed (1) hide show
  1. main.py +6 -6
main.py CHANGED
@@ -116,11 +116,11 @@ def format_model_info(metadata_list):
116
 
117
  # Créer une version formatée des informations des modèles
118
  formatted_info = ""
119
- for i, model in enumerate(metadata_list):
120
- formatted_info += f"### Modèle {i+1}\n"
121
- formatted_info += f"**ID:** {model.get('model_id', 'N/A')}\n"
122
- formatted_info += f"**Fichier:** {model.get('hf_filename', 'N/A')}\n"
123
- formatted_info += f"**Dernière mise à jour:** {model.get('updated_at', 'N/A')}\n\n"
124
 
125
  return formatted_info
126
 
@@ -186,7 +186,7 @@ app = mount_gradio_app(
186
  # We call init_models inside startup
187
  @app.on_event("startup")
188
  async def startup():
189
- """Initialiser l'API : charger les modèles depuis la base de données."""
190
  await init_models() # Call the consolidated init function
191
 
192
  @app.get("/health")
 
116
 
117
  # Créer une version formatée des informations des modèles
118
  formatted_info = ""
119
+ for model in metadata_list:
120
+ # Utiliser le nom du fichier comme titre principal
121
+ formatted_info += f"### Modèle : {model.get('hf_filename', 'N/A')}\n"
122
+ # Mettre l'ID et la dernière mise à jour sur une même ligne
123
+ formatted_info += f"**ID:** {model.get('model_id', 'N/A')} | **Dernière mise à jour:** {model.get('updated_at', 'N/A')}\n\n"
124
 
125
  return formatted_info
126
 
 
186
  # We call init_models inside startup
187
  @app.on_event("startup")
188
  async def startup():
189
+ """Initialiser l'API : charger les modèles depuis la base de données."""
190
  await init_models() # Call the consolidated init function
191
 
192
  @app.get("/health")