alexfremont commited on
Commit
078f3c3
·
1 Parent(s): 6ba6dc7

Add memory status display with auto-refresh every 10 seconds

Browse files
Files changed (1) hide show
  1. main.py +12 -0
main.py CHANGED
@@ -179,8 +179,20 @@ with gradio_app:
179
  outputs=[markdown_output, model_table, json_output, status]
180
  )
181
 
 
 
 
 
 
 
 
 
 
182
  # Initialiser les affichages au chargement
183
  gradio_app.load(fn=update_all_displays, outputs=[markdown_output, model_table, json_output, status])
 
 
 
184
 
185
  # Monter l'application Gradio à la racine dans FastAPI
186
  app = mount_gradio_app(
 
179
  outputs=[markdown_output, model_table, json_output, status]
180
  )
181
 
182
+ # Initialiser l'affichage de la mémoire dès le démarrage
183
+ memory_status = get_memory_status(model_pipelines)
184
+ status.value = format_memory_status(memory_status)
185
+
186
+ # Fonction dédiée pour mettre à jour uniquement le statut (informations mémoire)
187
+ def update_memory_status():
188
+ memory_status = get_memory_status(model_pipelines)
189
+ return format_memory_status(memory_status)
190
+
191
  # Initialiser les affichages au chargement
192
  gradio_app.load(fn=update_all_displays, outputs=[markdown_output, model_table, json_output, status])
193
+
194
+ # Mettre à jour le statut toutes les 10 secondes
195
+ status.update(every=10, callback=update_memory_status)
196
 
197
  # Monter l'application Gradio à la racine dans FastAPI
198
  app = mount_gradio_app(