Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -101,20 +101,20 @@ def generate_random_name(interpreter, vocab_size, sp, max_length=10, temperature
|
|
101 |
|
102 |
def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
|
103 |
hate_speech = detect_hate_speech(seed_text)
|
104 |
-
|
105 |
-
|
106 |
|
107 |
-
|
108 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
109 |
seed_text = ''
|
110 |
-
|
111 |
-
if hate_speech == ['Hate Speech']:
|
112 |
-
gr.Warning('Hate speech detected in the seed text, using an empty seed text.')
|
113 |
-
seed_text = ''
|
114 |
-
elif hate_speech == ['Offensive Speech']:
|
115 |
-
gr.Warning('Offensive speech detected in the seed text, using an empty seed text.')
|
116 |
-
seed_text = ''
|
117 |
-
# elif hate_speech == ['No Hate and Offensive Speech']:
|
118 |
|
119 |
if type == "Terraria":
|
120 |
max_seq_len = 12 # For skyrim = 13, for terraria = 12
|
|
|
101 |
|
102 |
def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
|
103 |
hate_speech = detect_hate_speech(seed_text)
|
104 |
+
profanity = detect_profanity([seed_text], language='All')
|
105 |
+
output = ''
|
106 |
|
107 |
+
if profanity > 0:
|
108 |
+
gr.Warning("Profanity detected in the seed text, using an empty seed text.")
|
109 |
+
seed_text = ''
|
110 |
+
else:
|
111 |
+
if hate_speech == ['Hate Speech']:
|
112 |
+
gr.Warning('Hate speech detected in the seed text, using an empty seed text.')
|
113 |
+
seed_text = ''
|
114 |
+
elif hate_speech == ['Offensive Speech']:
|
115 |
+
gr.Warning('Offensive speech detected in the seed text, using an empty seed text.')
|
116 |
seed_text = ''
|
117 |
+
# elif hate_speech == ['No Hate and Offensive Speech']:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
118 |
|
119 |
if type == "Terraria":
|
120 |
max_seq_len = 12 # For skyrim = 13, for terraria = 12
|