Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from gradio_client import Client
-import threading
 
 # Initialize clients
 client1 = Client("orionai/training-data-collection_2")
@@ -45,19 +44,18 @@ def update_dashboard():
         print(f"Error in update dashboard: {e}")
         return 0, 0
 
+def refresh_metrics():
+    tokens, speed = update_dashboard()
+    return tokens, speed
+
 # Create Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# Token Count Dashboard")
     
-    total_tokens = gr.Number(label="Total Token Count", value=0)
-    growth_speed = gr.Number(label="Total Growth Speed", value=0)
-    
-    def refresh_metrics():
-        tokens, speed = update_dashboard()
-        total_tokens.update(value=tokens)
-        growth_speed.update(value=speed)
+    total_tokens = gr.Number(label="Total Token Count")
+    growth_speed = gr.Number(label="Total Growth Speed")
     
     update_button = gr.Button("Update Stats")
-    update_button.click(refresh_metrics)
+    update_button.click(fn=refresh_metrics, inputs=[], outputs=[total_tokens, growth_speed])
     
 demo.launch()