charleselena commited on
Commit
e75aeb2
verified
1 Parent(s): b674557
Files changed (1) hide show
  1. handler.py +2 -3
handler.py CHANGED
@@ -8,6 +8,7 @@ from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
8
  from transformers import AutoProcessor, SafetyChecker
9
  #from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
10
 
 
11
 
12
  import torch
13
 
@@ -138,9 +139,7 @@ class EndpointHandler():
138
  safety_features = processor(image)
139
  safety_check_result = safety_checker(images=image, features=safety_features)
140
 
141
- print(f'Ocurri贸 un error: {safety_check_result}')
142
- print(f'Ocurri贸 un error: {safety_check_result["nsfw_content_detected"]')
143
-
144
 
145
  # run inference pipeline
146
  out = self.pipe(
 
8
  from transformers import AutoProcessor, SafetyChecker
9
  #from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
10
 
11
+ import pprint
12
 
13
  import torch
14
 
 
139
  safety_features = processor(image)
140
  safety_check_result = safety_checker(images=image, features=safety_features)
141
 
142
+ pprint(safety_check_result)
 
 
143
 
144
  # run inference pipeline
145
  out = self.pipe(