Update app.py

app.py

@@ -49,17 +49,12 @@ def generate(prompt, option, progress=gr.Progress()):
 
     # Safety check.
     feature_extractor_input = image_processor.postprocess(results.images, output_type="pil")
-    print("A")
     safety_checker_input = feature_extractor(feature_extractor_input, return_tensors="pt")
-    print("B")
     pixel_values = safety_checker_input.pixel_values.to(device, dtype)
-    print("C")
     images, has_nsfw_concept = safety_checker(
         images=results.images, clip_input=pixel_values
     )
-    print("D")
     if has_nsfw_concept[0]:
-        gr.Warning("Safety checker triggered. Image may contain violent or sexual content.")
         print(f"Safety checker triggered on prompt: {prompt}")
     return images[0]