Update app.py

app.py

@@ -101,10 +101,12 @@ with gr.Blocks(theme="ParityError/[email protected]") as demo:
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
     # Immediately load models
+    print("Loading models...")
     get_tinystyle_model(device)
     get_style_embedding_model(device)
     get_luar_model(device)
     get_mis_model(device)
+    print("Done loading models.")
 
     gr.Markdown("# TinyStyler Demo")
     gr.Markdown(