Skip to content

Commit c0e61cf

Browse files
committed
fix: [images_engine] fix reponse dict
1 parent 4d62b24 commit c0e61cf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

bin/lib/images_engine.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -221,12 +221,12 @@ def check_is_image_csam(obj_gid, image_description=False):
221221
# TODO LOG
222222
return {"status": "error", "reason": f"ollama requests error: {res.status_code}, {res.text}"}, 400
223223
else:
224-
r = res.json()
224+
r = res.json()['response']
225225
print(r)
226226
if r:
227-
if 'YES' in res:
227+
if 'YES' in r:
228228
is_csam = True
229-
elif 'NO' in res:
229+
elif 'NO' in r:
230230
is_csam = False
231231

232232
# TODO LOG NONE result

0 commit comments

Comments
 (0)