Update app.py

app.py

@@ -53,7 +53,7 @@ class ModelManager:
 
 model_manager = ModelManager()
 
-def get_system_stats():
+def get_system_stats(request: gr.Request = None):
     """Returns a dictionary of current system metrics with formatted strings."""
     mem = psutil.virtual_memory()
     disk = psutil.disk_usage('/')
@@ -61,7 +61,7 @@ def get_system_stats():
         f"CPU\t\t: {psutil.cpu_percent(interval=1)}%\n"
         f"Mem\t\t: {round(mem.used / (1024**3), 2)} / {round(mem.total / (1024**3), 2)} GB\n"
         f"Disk\t\t: {round(disk.used / (1024**3), 2)} / {round(disk.total / (1024**3), 2)} GB\n"
-        f"Active\t\t: {len(ACTIVE_SESSIONS)}/5min"
+        f"Active\t\t: {len(ACTIVE_SESSIONS) if request is None else live_count(request)}/5min"
     )
 
 def load_new_model(model_id):
@@ -208,7 +208,6 @@ with gr.Blocks(title="Small MF Model Tester", theme=gr.themes.Soft()) as app:
     )
     
     clean_btn.click(fn=clean_cache, outputs=[status_output])
-    app.load(fn=live_count, inputs=[], outputs=[], every=10)
 
 if __name__ == "__main__":
     app.launch()