Update app.py

app.py

@@ -1,7 +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 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
@@ -66,12 +66,12 @@ if 'beat' not in st.session_state:
 logger.info(f"Measuring memory before load_emo_model")
 print_memory_info()
 emotion_service = load_emo_model()
-# logger.info(f"Measuring memory before load_genre_model")
-# print_memory_info()
-# genre_service = load_genre_model()
-# logger.info(f"Measuring memory before load_beat_model")
-# print_memory_info()
-# beat_service = load_beat_model()
+logger.info(f"Measuring memory before load_genre_model")
+print_memory_info()
+genre_service = load_genre_model()
+logger.info(f"Measuring memory before load_beat_model")
+print_memory_info()
+beat_service = load_beat_model()
 logger.info(f"Measuring memory before load_image_model")
 print_memory_info()
 image_service = load_image_model()