nsfw_detect: Tolerate score computation failure

This commit is contained in:
Mia Herkt 2022-11-29 13:31:35 +01:00
parent af4b3b06c0
commit f25619b7e3
No known key found for this signature in database
GPG key ID: 72E154B8622EC191

View file

@ -77,10 +77,10 @@ class NSFWDetector:
"-cpng", "-i", fpath "-cpng", "-i", fpath
], stdout=PIPE, stderr=DEVNULL, check=True) ], stdout=PIPE, stderr=DEVNULL, check=True)
image_data = ff.stdout image_data = ff.stdout
except:
return -1.0
scores = self._compute(image_data) scores = self._compute(image_data)
except:
return -1.0
return scores[1] return scores[1]