pandora-s commited on
Commit
378b82f
·
verified ·
1 Parent(s): 0f5a036

Update tts.py

Browse files
Files changed (1) hide show
  1. tts.py +5 -5
tts.py CHANGED
@@ -22,11 +22,11 @@ def synthesize_and_save_audio(
22
  inputs=[input_text]
23
  )
24
 
25
- if moderation_response.results[0].categories.sexual or \
26
- moderation_response.results[0].categories.hate_and_discrimination or \
27
- moderation_response.results[0].categories.violence_and_threats or \
28
- moderation_response.results[0].categories.selfharm or \
29
- moderation_response.results[0].categories.jailbreaking:
30
  print("Input text blocked by moderation layer.", file=sys.stderr)
31
  return 3
32
 
 
22
  inputs=[input_text]
23
  )
24
 
25
+ if moderation_response.results[0].categories["sexual"] or \
26
+ moderation_response.results[0].categories["hate_and_discrimination"] or \
27
+ moderation_response.results[0].categories["violence_and_threats"] or \
28
+ moderation_response.results[0].categories["selfharm"] or \
29
+ moderation_response.results[0].categories["jailbreaking"]:
30
  print("Input text blocked by moderation layer.", file=sys.stderr)
31
  return 3
32