Update app.py

app.py

@@ -3,13 +3,13 @@ import torch
 import numpy as np
 import modin.pandas as pd
 from PIL import Image
-#from diffusers import DiffusionPipeline
+from diffusers import DiffusionPipeline
 from diffusers.models import AutoencoderKL
-from diffusers import StableDiffusionPipeline
+#from diffusers import StableDiffusionPipeline
 
 model = "prompthero/openjourney-v4"
 vae = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae")
-pipe = StableDiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16, safety_checker=None)
+pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16, safety_checker=None)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 #pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v4", torch_dtype=torch.float16, safety_checker=None)