Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 
 pipe = StableDiffusionPipeline.from_pretrained( #Downloading the model and naming it "pipe"
     "Lykon/dreamshaper-8", #Name of the model we will download
-    torch_dtype=torch.float16,) #Forces it to download the small version which saves time and computer memory
+    requires_safety_checker=False, safety_checker=None) #Disables the safety checker
 
 pipe.to("cuda") #Move the model to the GPU because it's faster than the CPU