Update

app.py

@@ -2,32 +2,27 @@
 
 from __future__ import annotations
 
+import os
+
 import gradio as gr
-import pandas as pd
+from apscheduler.schedulers.background import BackgroundScheduler
+from huggingface_hub import HfApi
 
 from demo_list import DemoList
 
-demo_list = DemoList()
-
-DEFAULT_COLUMNS = [
-    'title',
-    'arxiv',
-    'likes',
-    'tags',
-    'status',
-    'hardware',
-]
+INTERVAL_MIN = int(os.getenv('INTERVAL_MIN', '10'))
 
+demo_list = DemoList()
+api = HfApi()
 
-def update_df() -> pd.DataFrame:
-    demo_list.update_data()
-    return demo_list.df
-
+scheduler = BackgroundScheduler()
+scheduler.add_job(func=lambda: api.restart_space('hysts/list-of-demos'),
+                  trigger='interval',
+                  seconds=60 * INTERVAL_MIN)
+scheduler.start()
 
 with gr.Blocks(css='style.css') as demo:
     df = gr.Dataframe(value=demo_list.df,
                       datatype=demo_list.column_datatype,
                       type='pandas')
-    refresh_button = gr.Button('Refresh')
-    refresh_button.click(fn=update_df, outputs=df)
 demo.queue(api_open=False).launch()

demo_list.py

@@ -1,7 +1,6 @@
 import copy
 import datetime
 import operator
-import os
 import pathlib
 
 import pandas as pd
@@ -11,8 +10,6 @@ from huggingface_hub import HfApi
 
 repo_dir = pathlib.Path(__file__).parent
 
-HF_TOKEN = os.getenv('HF_TOKEN')
-
 
 class DemoList:
     COLUMN_INFO = [
@@ -31,7 +28,7 @@ class DemoList:
     ]
 
     def __init__(self):
-        self.api = HfApi(token=HF_TOKEN)
+        self.api = HfApi()
         self.data = self.load_data()
         self.df = self.to_df()
 
@@ -98,7 +95,7 @@ class DemoList:
         if status == 'RUNNING':
             color = 'green'
         elif status in ['STOPPED', 'PAUSED']:
-            color = 'yellow'
+            color = 'orange'
         elif status in ['RUNTIME_ERROR', 'BUILD_ERROR']:
             color = 'red'
         else:
@@ -109,7 +106,7 @@ class DemoList:
         if sdk == 'gradio':
             color = 'orange'
         elif sdk == 'docker':
-            color = 'cyan'
+            color = 'deepskyblue'
         else:
             color = ''
         return f'<div class="{color}">{sdk}</div>'
@@ -121,7 +118,7 @@ class DemoList:
         if 't4' in hardware or 'a10g' in hardware or 'a100' in hardware:
             color = 'red'
         elif suggested_hardware and 'cpu' not in suggested_hardware:
-            color = 'blue'
+            color = 'deepskyblue'
         else:
             return hardware
         return f'<div class="{color}">{hardware}</div>'

requirements.txt

@@ -1,3 +1,4 @@
+apscheduler==3.10.1
 gradio==3.36.1
 huggingface-hub==0.16.4
 pandas==2.0.2

style.css

@@ -22,16 +22,12 @@ div.green {
   color: green;
 }
 
-div.yellow {
-  color: yellow;
-}
-
 div.red {
   color: red;
 }
 
-div.blue {
-  color: blue;
+div.deepskyblue {
+  color: deepskyblue;
 }
 
 div.orange {