Update app.py

app.py

@@ -8,7 +8,7 @@ from huggingface_hub import login
 #import os
 #login(token=os.environ.get('HF_KEY'))
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = DiffusionPipeline.from_pretrained("dreamlike-art/dreamlike-photoreal-2.0", torch_dtype=torch.float16) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0")
+pipe = DiffusionPipeline.from_pretrained("segmind/SSD-1B", torch_dtype=torch.float16) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0")
 pipe = pipe.to(device)
 def resize(height, width, img):
     img = Image.open(img)