Update app.py

app.py

@@ -48,7 +48,20 @@ def load_image_model():
      pipeline.load_lora_weights("Weights/pytorch_lora_weights.safetensors", weight_name="pytorch_lora_weights.safetensors")
      return pipeline
 
+if 'emotion' not in st.session_state:
+    st.session_state.emotion = None
 
+if 'genre' not in st.session_state:
+    st.session_state.genre = None
+
+if 'beat' not in st.session_state:
+    st.session_state.beat = None
+
+logger.info(f"Measuring memory before load_emo_model")
+print_memory_info()
+emotion_service = load_emo_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()