Update app.py

app.py

@@ -22,13 +22,14 @@ OV_CONFIG = {
 }
 
 if BACKEND == "optimum":
-    from optimum.intel.openvino import OVDiffusionPipeline
+    from optimum.intel import OVDiffusionPipeline
 
     print("Loading with Optimum-Intel...")
     pipe =  OVDiffusionPipeline.from_pretrained(
         MODEL_PATH,
         device=DEVICE,
         ov_config=OV_CONFIG,
+        safety_checker=None
     )
 
     def generate(prompt, negative_prompt, num_steps, guidance_scale, seed):