Hugging Face's logo

Spaces:
EQUES
/
NGword-detector-ver2
Sleeping

App Files Community
kb2022 commited on
Commit
352aba6
·
verified ·
1 Parent(s): 006006e

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,10 +4,10 @@ from utils.find_ng_word import get_ng_wordlist, get_ng_wordlist_from_saved, sear
4
  from utils.llm import load_llm_from_pretrained, inference
5
 
6
 
7
- wordlist_1_path_s = "/content/drive/MyDrive/llm_qlora_ngword/ng_wordlists/ng_wordlist_1_sexual.txt"
8
- wordlist_2_path_s = "/content/drive/MyDrive/llm_qlora_ngword/ng_wordlists/ng_wordlist_2_sexual.txt"
9
- wordlist_1_path_o = "/content/drive/MyDrive/llm_qlora_ngword/ng_wordlists/ng_wordlist_1_offensive.txt"
10
- wordlist_2_path_o = "/content/drive/MyDrive/llm_qlora_ngword/ng_wordlists/ng_wordlist_2_offensive.txt"
11
 
12
  pretrained_model_path = "input/llm_weights"
13
 
 
4
  from utils.llm import load_llm_from_pretrained, inference
5
 
6
 
7
+ wordlist_1_path_s = "input/ng_wordlists/ng_wordlist_1_sexual.txt"
8
+ wordlist_2_path_s = "input/ng_wordlists/ng_wordlist_2_sexual.txt"
9
+ wordlist_1_path_o = "input/ng_wordlists/ng_wordlist_1_offensive.txt"
10
+ wordlist_2_path_o = "input/ng_wordlists/ng_wordlist_2_offensive.txt"
11
 
12
  pretrained_model_path = "input/llm_weights"
13