Update app.py

app.py

@@ -49,10 +49,15 @@ 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=safety_checker_input.pixel_values.to(device, dtype)
+        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}")