Update app.py

app.py

@@ -11,7 +11,7 @@ import spaces
 import torch
 from diffusers import DiffusionPipeline
 
-DESCRIPTION = """# Playground v2.5"""
+DESCRIPTION = """# Dall E 3 XL"""
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
 
@@ -23,11 +23,11 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-NUM_IMAGES_PER_PROMPT = 1
+NUM_IMAGES_PER_PROMPT = 4
 
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(
-        "playgroundai/playground-v2.5-1024px-aesthetic",
+        "ehristoforu/dalle-3-xl",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False,