psi-jack/0x0
1
0
Fork 0
mirror of https://git.0x0.st/mia/0x0.git synced 2024-11-09 11:48:59 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

nsfw_detect: Tolerate score computation failure

Browse source
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
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
nsfw_detect.py
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]