Update app.py

app.py

@@ -21,9 +21,8 @@ def print_memory_info():
     logger.info(f"Free GPU Memory: {free_gpu_mem:.2f} GB")
     
 
-# @st.cache_resource
+@st.cache_resource
 def load_image_model():
-     print_memory_info()
      torch.cuda.empty_cache()
      pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", variant='fp16')
      pipeline.to("cuda")
@@ -31,5 +30,6 @@ def load_image_model():
      return pipeline
 
 
+print_memory_info()
 image_service = load_image_model()
-
+print_memory_info()