Update app.py

app.py

@@ -18,9 +18,10 @@ from diffusers import DiffusionPipeline   # StableDiffusionPipeline, StableDiffu
 # Nvidia GPU config
 
 #from diffusers import DiffusionPipeline
-pipe = DiffusionPipeline.from_pretrained("mswhite/nnn-model")
-#pipe = StableDiffusionPipeline.from_pretrained("mswhite/nnn-model", torch_dtype=torch.float16) #, safety_checker=None)
-#pipe = pipe.to("cuda")
+#pipe = DiffusionPipeline.from_pretrained("mswhite/nnn-model")
+
+pipe = StableDiffusionPipeline.from_pretrained("mswhite/nnn-model", torch_dtype=torch.float16) #, safety_checker=None)
+pipe = pipe.to("cuda")
 pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
 
 # CPU config