Upload app.py

app.py

@@ -24,24 +24,17 @@ device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is
 if device == "mps":
     os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = "1"
 
-# Load model once at startup using caching
-@spaces.GPUCache
-def load_models():
-    model_id = "runwayml/stable-diffusion-v1-5"
-    
-    pipe = StableDiffusionPipeline.from_pretrained(
-        model_id,
+# Load model with proper caching
+@spaces.GPU
+def load_model():
+    return StableDiffusionPipeline.from_pretrained(
+        "runwayml/stable-diffusion-v1-5",
         torch_dtype=torch.float16,
-        safety_checker=None,
-        use_safetensors=True
+        safety_checker=None
     ).to(device)
-    
-    # Disable unnecessary progress bars
-    pipe.set_progress_bar_config(disable=True)
-    return pipe
 
-# Initialize pipeline once
-pipe = load_models()
+# Create pipeline instance
+pipe = load_model()
 
 # Load concept library
 concept_embeds, concept_tokens = load_concept_library(pipe)
@@ -317,4 +310,4 @@ def create_demo():
 # Launch the app
 if __name__ == "__main__":
     demo = create_demo()
-    demo.launch(debug=False, show_error=True, server_name="0.0.0.0", server_port=7860)
+    demo.launch(debug=False, show_error=True, server_name="0.0.0.0", server_port=7860, cache_examples=True)