Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 import importlib
 from functools import lru_cache
+from time import time
 
 # Import TinyStyler
 tinystyler_module = importlib.util.module_from_spec(
@@ -51,6 +52,7 @@ def run_tinystyler(source_text, target_texts, reranking, temperature, top_p):
             200,
             device=device,
             seed=42,
+            log_timing=True,
         )[0]
     except Exception as e:
         return f"Error: {e}"
@@ -101,12 +103,12 @@ with gr.Blocks(theme="ParityError/Interstellar@0.0.1") as demo:
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
     # Immediately load models
-    print("Loading models...")
+    print("Loading models...", time())
     get_tinystyle_model(device)
     get_style_embedding_model(device)
     get_luar_model(device)
     get_mis_model(device)
-    print("Done loading models.")
+    print("Done loading models.", time())
 
     gr.Markdown("# TinyStyler Demo")
     gr.Markdown(
@@ -164,10 +166,10 @@ with gr.Blocks(theme="ParityError/Interstellar@0.0.1") as demo:
     )
 
     # Pre-warm all examples
-    # print("Pre-warming...")
+    # print("Pre-warming...", time())
     # for preset_example in preset_examples.values():
     #     run_tinystyler("A test.", preset_example["target_texts"], 1, 1.0, 1.0)
-    # print("Done pre-warming.")
+    # print("Done pre-warming.", time())
 
     # Initialize the fields with the first example
     (