Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
078f3c3
1
Parent(s):
6ba6dc7
Add memory status display with auto-refresh every 10 seconds
Browse files
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(
|