Update app.py

app.py

@@ -57,11 +57,17 @@ if 'genre' not in st.session_state:
 if 'beat' not in st.session_state:
     st.session_state.beat = None
 
-logger.info(f"Measuring memory before load_emo_model")
+# 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()
-emotion_service = load_emo_model()
+beat_service = load_beat_model()
 logger.info(f"Measuring memory before load_image_model")
 print_memory_info()
-# image_service = load_image_model()
-# logger.info(f"Measuring memory after load_image_model")
-# print_memory_info()
+image_service = load_image_model()
+logger.info(f"Measuring memory after load_image_model")
+print_memory_info()