Update app.py

app.py

@@ -1,4 +1,4 @@
-from diffusers import StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline, EulerAncestralDiscreteScheduler
+from diffusers import AutoPipelineForText2Image, EulerAncestralDiscreteScheduler
 from diffusers import UniPCMultistepScheduler
 import gradio as gr
 import torch
@@ -19,7 +19,7 @@ auth_token=os.environ["ACCESS_TOKEN"]
 
 scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler",use_auth_token=auth_token)
 
-pipe = StableDiffusionXLPipeline.from_pretrained(
+pipe = AutoPipelineForText2Image.from_pretrained(
   model_id,
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
   scheduler=scheduler,