Shows cache status on startup.

app.py

@@ -62,6 +62,16 @@ MODEL_MAP = {
 }
 
 
+def get_cache_status():
+  """Returns a string summarizing cache status."""
+  mem_n, mem_sz = gradio_helpers.get_memory_cache_info()
+  disk_n, disk_sz = gradio_helpers.get_disk_cache_info()
+  return (
+      f'memory cache {mem_n} items [{mem_sz/1e9:.2f}G], '
+      f'disk cache {disk_n} items [{disk_sz/1e9:.2f}G]'
+  )
+
+
 def compute(image_path, prompts, family, variant, res, bias, progress=gr.Progress()):
   """Loads model and computes answers."""
 
@@ -117,12 +127,8 @@ def compute(image_path, prompts, family, variant, res, bias, progress=gr.Progres
   state = list(zip(prompts, [round(p.item(), 3) for p in text_probs]))
 
   dt = time.monotonic() - t0
-  mem_n, mem_sz = gradio_helpers.get_memory_cache_info()
-  disk_n, disk_sz = gradio_helpers.get_disk_cache_info()
   status = gr.Markdown(
-    f'Computed inference in {dt:.1f} seconds ('
-    f'memory cache {mem_n} items, {mem_sz/1e6:.1f} M, '
-    f'disk cache {disk_n} items, {disk_sz/1e6:.1f} M)')
+      f'Computed inference in {dt:.1f} seconds ({get_cache_status()})')
 
   if 'b' in out:
     logging.info('model_name=%s default bias=%f', model_name, out['b'])
@@ -155,7 +161,7 @@ def create_app():
 
     gr.Markdown('Gradio clone of the original [LiT demo](https://google-research.github.io/vision_transformer/lit/).')
 
-    status = gr.Markdown()
+    status = gr.Markdown(f'Ready ({get_cache_status()})')
 
     with gr.Row():
       image = gr.Image(label='Image', type='filepath')