Update app.py

app.py

@@ -1,4 +1,7 @@
 import streamlit as st
+from flask.Emotion_spotting_service import _Emotion_spotting_service
+from flask.Genre_spotting_service import _Genre_spotting_service
+from flask.Beat_tracking_service import _Beat_tracking_service
 from diffusers import StableDiffusionPipeline
 import torch
 import os
@@ -20,7 +23,23 @@ def print_memory_info():
     logger.info(f"Free CPU Memory: {free_cpu_mem:.2f} GB")
     logger.info(f"Free GPU Memory: {free_gpu_mem:.2f} GB")
     
+emo_list = []
+gen_list = []
+tempo_list = []
+@st.cache_resource
+def load_emo_model():
+    emo_service = _Emotion_spotting_service("flask/emotion_model.h5")
+    return emo_service
+@st.cache_resource
+def load_genre_model():
+    gen_service = _Genre_spotting_service("flask/Genre_classifier_model.h5")
+    return gen_service
 
+@st.cache_resource
+def load_beat_model():
+    beat_service = _Beat_tracking_service()
+    return beat_service
+    
 @st.cache_resource
 def load_image_model():
      torch.cuda.empty_cache()