Update app.py

app.py

@@ -9,7 +9,7 @@ selected_model = "timbrooks/instruct-pix2pix"
 
 @st.cache_resource
 def load_model():
-    return StableDiffusionInstructPix2PixPipeline.from_pretrained(selected_model).to(torch.float32).to(device)
+    return StableDiffusionInstructPix2PixPipeline.from_pretrained(selected_model, torch_dtype=torch.float32, safety_checker=None).to(device)
 
 pipe = load_model()