Update app.py

app.py

@@ -1,4 +1,4 @@
-from diffusers import AutoPipelineForText2Image, DiffusionPipeline, UniPCMultistepScheduler, EulerAncestralDiscreteScheduler
+from diffusers import AutoPipelineForText2Image, DiffusionPipeline, StableDiffusionPipeline, UniPCMultistepScheduler, EulerAncestralDiscreteScheduler
 import torch
 import gradio as gr
 from PIL import Image
@@ -10,7 +10,7 @@ from accelerate import Accelerator
 
 accelerator = Accelerator()
 apol=[]
-pipe = accelerator.prepare(AutoPipelineForText2Image.from_single_file("https://huggingface.co/lllyasviel/fav_models/fav/DreamShaper_8_pruned.safetensors",torch_dtype=torch.float32, variant=None, use_safetensors=True, safety_checker=None))
+pipe = accelerator.prepare(StableDiffusionPipeline.from_single_file("https://huggingface.co/lllyasviel/fav_models/fav/DreamShaper_8_pruned.safetensors",torch_dtype=torch.float32, variant=None, use_safetensors=True, safety_checker=None))
 ##pipe.scheduler = accelerator.prepare(EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config))
 pipe.unet.to(memory_format=torch.channels_last)
 pipe = accelerator.prepare(pipe.to("cpu"))