Update app.py

app.py

@@ -38,21 +38,6 @@ def generate_random_name(interpreter, vocab_size, sp, max_length=10, temperature
     decoded_name = ''
 
     if seed_text:
-        hate_speech = detect_hate_speech(seed_text)
-        profanity = detect_profanity([seed_text], language='All')
-        output = ''
-    
-        if profanity > 0:
-            gr.Info("Profanity detected in the seed text, using an empty seed text.")
-            seed_text = ''
-        else:
-            if hate_speech == ['Hate Speech']:
-                gr.Info('Hate Speech detected in the seed text, using an empty seed text.')
-                seed_text = ''
-            elif hate_speech == ['Offensive Speech']:
-                gr.Info('Offensive Speech detected in the seed text, using an empty seed text.')
-                seed_text = ''
-            # elif hate_speech == ['No Hate and Offensive Speech']:
         generated_name = seed_text
     else:
         random_index = np.random.randint(1, vocab_size)
@@ -115,6 +100,22 @@ def generate_random_name(interpreter, vocab_size, sp, max_length=10, temperature
     return generated_name.strip()
 
 def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
+    hate_speech = detect_hate_speech(seed_text)
+        profanity = detect_profanity([seed_text], language='All')
+        output = ''
+    
+        if profanity > 0:
+            gr.Warning("Profanity detected in the seed text, using an empty seed text.")
+            seed_text = ''
+        else:
+            if hate_speech == ['Hate Speech']:
+                gr.Warning('Hate speech detected in the seed text, using an empty seed text.')
+                seed_text = ''
+            elif hate_speech == ['Offensive Speech']:
+                gr.Warning('Offensive speech detected in the seed text, using an empty seed text.')
+                seed_text = ''
+            # elif hate_speech == ['No Hate and Offensive Speech']:
+    
     if type == "Terraria":
         max_seq_len = 12 # For skyrim = 13, for terraria = 12
         sp = spm.SentencePieceProcessor()