mia/0x0
1
0
Fork 1
mirror of https://git.0x0.st/mia/0x0.git synced 2024-11-24 17:37:12 +01:00
Code Issues Projects Releases Packages Wiki Activity

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
], stdout=PIPE, stderr=DEVNULL, check=True)
image_data = ff.stdout
except:
return -1.0
scores = self._compute(image_data)
except:
return -1.0
return scores[1]