Update app.py

app.py

@@ -16,7 +16,7 @@ from composable_diffusion.model_creation import create_model_and_diffusion as cr
 from composable_diffusion.model_creation import model_and_diffusion_defaults as model_and_diffusion_defaults_for_clevr
 
 from torch import autocast
-from diffusers import StableDiffusionPipeline
+from composable_stable_diffusion_pipeline import ComposableStableDiffusionPipeline
 
 # This notebook supports both CPU and GPU.
 # On CPU, generating one sample may take on the order of 20 minutes.
@@ -27,7 +27,7 @@ device = th.device('cpu' if not th.cuda.is_available() else 'cuda')
 print(device)
 
 # init stable diffusion model
-pipe = StableDiffusionPipeline.from_pretrained(
+pipe = ComposableStableDiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",
     use_auth_token=st.secrets["USER_TOKEN"]
 ).to(device)