Update app.py

app.py

@@ -4,6 +4,7 @@ os.environ["KERAS_BACKEND"] ="tensorflow"
 import random
 import keras
 import gradio as gr
+import time   # <-- added
 from src.inference import generate_chorale, draw_random_sample
 from src.dataset import NoteEncoder
 from src.metrics import Preplexity
@@ -21,6 +22,7 @@ wav_path = os.path.join(AUDIO_SAMPLES_PATH, "sample.wav")
 
 
 ### DOWNLOAD SF2 MUSIC FONT ###
+t0 = time.time()
 sf2_download_path = keras.utils.get_file(
         "FluidR3_GM.zip",
         "https://keymusician01.s3.amazonaws.com/FluidR3_GM.zip",
@@ -29,15 +31,23 @@ sf2_download_path = keras.utils.get_file(
         cache_subdir= ""
         )
 SF2_PATH = os.path.join(sf2_download_path, "FluidR3_GM.sf2")
+print(f"SF2 font download/extract took {time.time()-t0:.2f}s")
 
 
 ### LOAD MODEL & ENCODERS ###
+print("loading model...")
+t0 = time.time()
 model = keras.models.load_model(os.path.join(MODEL_PATH, "bach_model.keras"),
                                 custom_objects={"Preplexity": Preplexity})
+print(f"model loaded in {time.time()-t0:.2f}s")
+
+t1 = time.time()
 note2id, id2note, vocab = NoteEncoder(vocab_path=ARTIFACTS_PATH, samples_path=None)
+print(f"NoteEncoder init took {time.time()-t1:.2f}s")
 
 
 ### GRADIO ASSETS ###
+t2 = time.time()
 css = load_css()
 english_summary = load_markdown("english_summary")
 persian_summary = load_markdown("persian_summary")
@@ -45,6 +55,7 @@ english_help = load_markdown("english_help")
 persian_help = load_markdown("persian_help")
 english_title = "# BachNet: AI-Generated Bach Music"
 persian_title = "# باخ‌نت: خلق موسیقی مشابه باخ با هوش مصنوعی"
+print(f"Loaded CSS/markdown assets in {time.time()-t2:.2f}s")
 
 
 ### GENERATION FUNCTIONS ###
@@ -53,7 +64,9 @@ def pick_random_seed():
 
 def generate_fn(seed_path, seed_len, gen_len, temp):
     sample_rows = slice(0, seed_len)
-    
+
+    print("=== Generation started ===")
+    t0 = time.time()
     generate_chorale(
         model=model,
         sample_seed_path=seed_path,
@@ -64,8 +77,14 @@ def generate_fn(seed_path, seed_len, gen_len, temp):
         temperature=temp,
         sample_seed_rows=sample_rows
     )
-    
+    t1 = time.time()
+    print(f"generate_chorale took {t1 - t0:.2f}s")
+
     midi_to_wave(midi_file_path=midi_path, SF2_PATH=SF2_PATH, wave_path=wav_path)
+    t2 = time.time()
+    print(f"midi_to_wave took {t2 - t1:.2f}s")
+    print(f"Total generate_fn time {t2 - t0:.2f}s")
+
     return wav_path
 
 def set_english():
@@ -121,4 +140,4 @@ with gr.Blocks(css=css, title="BachNet") as demo:
 
 ### LAUNCH APP ###
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()