Upload app.py

app.py

@@ -20,8 +20,8 @@ model, tokenizer = load_llm_from_pretrained(pretrained_model_path)
 # 検出結果を生成
 def detect_ng_word(input_text):
     response = []
-    rtn_s = search_ng_word(data_point["input"], ng_wordlist_1_s, ng_wordlist_2_s)
-    rtn_o = search_ng_word(data_point["input"], ng_wordlist_1_o, ng_wordlist_2_o)
+    rtn_s = search_ng_word(input_text, ng_wordlist_1_s, ng_wordlist_2_s)
+    rtn_o = search_ng_word(input_text, ng_wordlist_1_o, ng_wordlist_2_o)
     rtn = rtn_s + rtn_o
 
     if len(rtn) == 0: