Spaces:
Runtime error
Runtime error
Commit
•
1aaf877
1
Parent(s):
b5d3ac5
Update inpainting.py
Browse files- inpainting.py +1 -9
inpainting.py
CHANGED
@@ -183,12 +183,4 @@ class StableDiffusionInpaintingPipeline(DiffusionPipeline):
|
|
183 |
|
184 |
image = (image / 2 + 0.5).clamp(0, 1)
|
185 |
image = image.cpu().permute(0, 2, 3, 1).numpy()
|
186 |
-
|
187 |
-
# run safety checker
|
188 |
-
safety_cheker_input = self.feature_extractor(self.numpy_to_pil(image), return_tensors="pt").to(self.device)
|
189 |
-
image, has_nsfw_concept = self.safety_checker(images=image, clip_input=safety_cheker_input.pixel_values)
|
190 |
-
|
191 |
-
if output_type == "pil":
|
192 |
-
image = self.numpy_to_pil(image)
|
193 |
-
|
194 |
-
return {"sample": image, "nsfw_content_detected": has_nsfw_concept}
|
|
|
183 |
|
184 |
image = (image / 2 + 0.5).clamp(0, 1)
|
185 |
image = image.cpu().permute(0, 2, 3, 1).numpy()
|
186 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|