OzoneAsai commited on
Commit
e63359b
1 Parent(s): 4e2a958

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -72,7 +72,8 @@ def blur_image(image: Image.Image, blur_enabled: bool, nsfw_score: float) -> Ima
72
  nsfw_result = predict_nsfw(image)
73
  nsfw_score = nsfw_result.get('nsfw', 0.0)
74
  if nsfw_score >= 0.75:
75
- image = image.filter(ImageFilter.BLUR)
 
76
  return image
77
 
78
  # ページングと並び替えのためのデータを取得する関数
 
72
  nsfw_result = predict_nsfw(image)
73
  nsfw_score = nsfw_result.get('nsfw', 0.0)
74
  if nsfw_score >= 0.75:
75
+ blur_radius=20*nsfw_score
76
+ image = image.filter(ImageFilter.BLUR(radius=blur_radius))
77
  return image
78
 
79
  # ページングと並び替えのためのデータを取得する関数