Update app.py

app.py

@@ -55,7 +55,7 @@ def generate(
     lora = 'amazonaws-la/juliette',
 ) -> PIL.Image.Image:
     if torch.cuda.is_available():
-        pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16, safety_checker=None)
+        pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
 
         if use_lora:
             pipe.load_lora_weights(lora)