Uncommented load image model function

app.py

@@ -22,11 +22,11 @@ def load_beat_model():
     beat_service = _Beat_tracking_service()
     return beat_service
 
-#@st.cache_resource
-# def load_image_model():
-#      pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5",torch_dtype=torch.float16).to("cuda")
-#      pipeline.load_lora_weights("Weights/pytorch_lora_weights.safetensors", weight_name="pytorch_lora_weights.safetensors")
-#      return pipeline
+@st.cache_resource
+def load_image_model():
+     pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5",torch_dtype=torch.float16).to("cuda")
+     pipeline.load_lora_weights("Weights/pytorch_lora_weights.safetensors", weight_name="pytorch_lora_weights.safetensors")
+     return pipeline
 
 
 if 'emotion' not in st.session_state: