Update app.py

app.py

@@ -18,7 +18,7 @@ import numpy as np
 from PIL import Image
 import spaces
 import torch
-from diffusers import DiffusionPipeline
+from diffusers import DiffusionPipeline, StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 from typing import Tuple
 
 #BaseConditions--
@@ -246,7 +246,7 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 #Compile
 if torch.cuda.is_available():
- pipe = DiffusionPipeline.from_pretrained(
+ pipe = StableDiffusionXLPipeline.from_pretrained(
  "SG161222/RealVisXL_V4.0",
  torch_dtype=torch.float16,
  use_safetensors=True,