Spaces:

momo
/

Hate_speech_Privacy_Detection

Running

momo commited on Jun 6, 2022

Commit

ec82774

•

1 Parent(s): c21f82a

add app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import TextClassificationPipeline
 import gradio as gr
 # global var
-MODEL_NAME = 'momo/KcBERT-base_Hate_speech_Privacy_Detection'
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSequenceClassification.from_pretrained(
     MODEL_NAME,

 import gradio as gr
 # global var
+MODEL_NAME = 'momo/KcELECTRA-base_Hate_speech_Privacy_Detection'
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSequenceClassification.from_pretrained(
     MODEL_NAME,